在使用数据生成器(如来自数据帧的流)训练模型时,如何获取roc_auc_score

时间:2019-06-04 03:43:41

标签: keras scikit-learn sklearn-pandas

利用sklearn-metrics.roc_auc_score和sklearn-metrics.roc_curve中roc_curve的数据来生成roc_auc_score,而我正在使用诸如flow_dataframe之类的数据生成器来训练模型。

from keras.preprocessing.image import ImageDataGenerator
data_gen3 = ImageDataGenerator(
    rescale=1./255,
    shear_range=0.2,
    zoom_range=0.2,
    rotation_range=20,
    width_shift_range=0.2,
    height_shift_range=0.2,
    horizontal_flip=True)

0 个答案:

没有答案