Skip to content
Snippets Groups Projects
Commit 609825a3 authored by Schneider Leo's avatar Schneider Leo
Browse files

fix : make_prediction

parent edb9f7b5
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
Showing
with 2 additions and 2 deletions
image_ref/main.py
+ 2
− 2
View file @ 609825a3
...@@ -175,8 +175,8 @@ def make_prediction_duo(model, data, f_name, f_name2): ...@@ -175,8 +175,8 @@ def make_prediction_duo(model, data, f_name, f_name2):
for i in range(n_class): for i in range(n_class):
confidence_matrix[i]=np.mean(confidence_pred_list[i],axis=0) confidence_matrix[i]=np.mean(confidence_pred_list[i],axis=0)
df_cm = pd.DataFrame(cf_matrix / np.sum(cf_matrix, axis=1)[:, None], index=[i for i in range(2)], df_cm = pd.DataFrame(cf_matrix / np.sum(cf_matrix, axis=1)[:, None], index=[i for i in classes],
columns=['True','False']) columns=[i for i in classes])
print('Saving Confusion Matrix') print('Saving Confusion Matrix')
plt.clf() plt.clf()
plt.figure(figsize=(14, 9)) plt.figure(figsize=(14, 9))
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment