我就废话不多说了,大家还是直接看代码吧!
# 利用sklearn自建评价函数
from sklearn.model_selection import train_test_split
from sklearn.metrics import roc_auc_score
from keras.callbacks import Callback
class RocAucEvaluation(Callback):
def __init__(self, validation_data=(), interval=1):
super(Callback, self).__init__()
self.i
2022-02-09 19:50:36
40KB
ar
AS
le
1