Commit 26b85e7e authored by Jana Schor's avatar Jana Schor
Browse files

test on 1st target only removed

parent b6fb811d
......@@ -1233,6 +1233,7 @@ def trainNNmodelsMulti(modelfilepathprefix, x, y, split=0.2, epochs=500, params=
if 'id' in y.columns:
y = y.drop('id', axis=1)
# drop compounds that are not measured for all target columns, transform to numpy
(xmulti, ymulti) = shuffleDataPriorToTraining(x, y)
......@@ -1537,6 +1538,7 @@ def trainNNmodels(modelfilepathprefix, x, y, split=0.2, epochs=50, params=None,
allscores = allscores.append(row_df, ignore_index=True)
fold_no = fold_no + 1
del model
# now next fold
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