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(): ...@@ -67,8 +67,10 @@ def main():
67 print("Model loaded") 67 print("Model loaded")
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 arguments["--reject"]:
71 - y[np.absolute(proba[:,0] - proba[:,1]) < float(arguments["--reject"])] = -1 71 + diff = np.absolute(proba[:,0] - proba[:,1])
72 + thresh = float(arguments["--reject"])
73 + y[diff < thresh] = -1
72 bmu = np.array(som.get_BMUS(data)) 74 bmu = np.array(som.get_BMUS(data))
73 75
74 save_pred(som,data,data_names,y,proba,bmu,output_path) 76 save_pred(som,data,data_names,y,proba,bmu,output_path)
......