Ludovic PLATON / IRSOM

Go to a project
Toggle navigation Toggle navigation pinning
Ludovic PLATON
0ad1c305 1 parent 6c514cfa

Working rejection

Inline Side-by-side
Showing 1 changed file with 4 additions and 2 deletions
......@@ -67,8 +67,10 @@ def main():
print("Model loaded")
y,proba = ssom.predict(data)
y = np.array(y)
if not arguments["--reject"]:
y[np.absolute(proba[:,0] - proba[:,1]) < float(arguments["--reject"])] = -1
if arguments["--reject"]:
diff = np.absolute(proba[:,0] - proba[:,1])
thresh = float(arguments["--reject"])
y[diff < thresh] = -1
bmu = np.array(som.get_BMUS(data))
save_pred(som,data,data_names,y,proba,bmu,output_path)
......