Commit d597d86c by Pamela Osuna

takes the better parameters

parent b472c653
......@@ -27,5 +27,11 @@ combinations = itertools.product(c_,b_,e_)
input_, output_ = parser.parse_data(n_experiences)
#run an specific combination
max_params = combinations[0]
max_avg_auc = 0
for params in combinations:
cnn.run_nn(input_, output_, n_experiences, params)
avg_acc, avg_auc = cnn.run_nn(input_, output_, n_experiences, params)
if avg_auc > max_avg_auc:
max_avg_auc = avg_auc
max_params = params
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment