Confusion matrix_report: cm_report(test_g)
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
Comments