Ludovic PLATON / IRSOM

Go to a project
Toggle navigation Toggle navigation pinning
Ludovic PLATON
6c514cfa 1 parent 646b068c

Fix reject

Inline Side-by-side
Showing 1 changed file with 1 additions and 1 deletions
...@@ -68,7 +68,7 @@ def main(): ...@@ -68,7 +68,7 @@ def main():
68 y,proba = ssom.predict(data) 68 y,proba = ssom.predict(data)
69 y = np.array(y) 69 y = np.array(y)
70 if not arguments["--reject"]: 70 if not arguments["--reject"]:
71 - y[proba < float(arguments["--reject"])] = -1 71 + y[np.absolute(proba[:,0] - proba[:,1]) < float(arguments["--reject"])] = -1
72 bmu = np.array(som.get_BMUS(data)) 72 bmu = np.array(som.get_BMUS(data))
73 73
74 save_pred(som,data,data_names,y,proba,bmu,output_path) 74 save_pred(som,data,data_names,y,proba,bmu,output_path)
......