Confusion matrix_report: cm_report(test_g)

PHOTO EMBED

Wed Aug 03 2022 12:00:41 GMT+0000 (Coordinated Universal Time)

Saved by @mnis00014

def cm_report(df):
    num_label = {'ok': 1, 'nok' : 0}
    Y_test = df_test['class'].copy().map(num_label).astype('int')

    df.reset()
    predictions = model.predict(df, steps=len(df), verbose=0)
    pred_labels= np.where(predictions>0.5, 1, 0)

    target_names = ['nok : 0', 'ok : 1']
    print(classification_report(Y_test, pred_labels, target_names = target_names))
content_copyCOPY