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

fix : make_prediction

parent 70c64d51
No related branches found
No related tags found
No related merge requests found
......@@ -164,10 +164,13 @@ def make_prediction_duo(model, data, f_name, f_name2):
confidence_pred_list[specie].append(output[:,0].data.cpu().numpy())
#Mono class output (only most postive paire)
output = torch.argmax(output[:,0])
label = torch.argmin(label)
print(output, label)
y_pred.extend(output)
label = torch.argmin(label)
print(output, label)
y_true.extend(label) # Save Truth
# constant for classes
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment