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():
y,proba = ssom.predict(data)
y = np.array(y)
if not arguments["--reject"]:
y[proba < float(arguments["--reject"])] = -1
y[np.absolute(proba[:,0] - proba[:,1]) < float(arguments["--reject"])] = -1
bmu = np.array(som.get_BMUS(data))
save_pred(som,data,data_names,y,proba,bmu,output_path)
......