Commit b2a3b50d authored by lucas_miranda's avatar lucas_miranda
Browse files

Changed epochs default for model training

parent 3a9eca76
Pipeline #99995 passed with stages
in 22 minutes and 10 seconds
......@@ -458,7 +458,7 @@ def autoencoder_fitting(
if phenotype_prediction > 0.0:
ys += [y_train[-Xs.shape[0] :, 0]]
yvals += [y_val[-Xs.shape[0] :, 0]]
yvals += [y_val[-Xvals.shape[0] :, 0]]
# Remove the used column (phenotype) from both y arrays
y_train = y_train[:, 1:]
......@@ -466,7 +466,7 @@ def autoencoder_fitting(
if rule_based_prediction > 0.0:
ys += [y_train[-Xs.shape[0] :]]
yvals += [y_val[-Xs.shape[0] :]]
yvals += [y_val[-Xvals.shape[0] :]]
ae.fit(
x=Xs,
......@@ -633,16 +633,16 @@ def tune_search(
Xvals, yvals = X_val[:-1], [X_val[:-1], X_val[1:]]
if phenotype_prediction > 0.0:
ys += [y_train[:, 0]]
yvals += [y_val[:, 0]]
ys += [y_train[-Xs.shape[0]:, 0]]
yvals += [y_val[-Xvals.shape[0]:, 0]]
# Remove the used column (phenotype) from both y arrays
y_train = y_train[:, 1:]
y_val = y_val[:, 1:]
if rule_based_prediction > 0.0:
ys += [y_train]
yvals += [y_val]
ys += [y_train[-Xs.shape[0]:]]
yvals += [y_val[-Xvals.shape[0]:]]
tuner.search(
Xs,
......
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