2020
10-08
10-08
Keras 利用sklearn的ROC-AUC建立评价函数详解
我就废话不多说了,大家还是直接看代码吧!#利用sklearn自建评价函数fromsklearn.model_selectionimporttrain_test_splitfromsklearn.metricsimportroc_auc_scorefromkeras.callbacksimportCallbackclassRocAucEvaluation(Callback):def__init__(self,validation_data=(),interval=1):super(Callback,self).__init__()self.interval=intervalself.x_val,self.y_val=validation_datadefon_epoch_end(self,...
继续阅读 >