Commit b46030d0 authored by CCXD's avatar CCXD Committed by GitHub
Browse files

Update chembl_tf_models.py

Small fix
parent 3fb43dc0
Loading
Loading
Loading
Loading
+1 −2
Original line number Diff line number Diff line
"""
Script that trains Tensorflow Multitask models on KAGGLE datasets.
Script that trains Tensorflow Multitask models on ChEMBL dataset.
"""

from __future__ import print_function
@@ -51,7 +51,6 @@ model.fit(train_dataset, nb_epoch=nb_epoch)

train_scores = model.evaluate(train_dataset, [metric], transformers)
valid_scores = model.evaluate(valid_dataset, [metric], transformers)
#Only use for final evaluation
test_scores = model.evaluate(test_dataset, [metric], transformers)

print("Train scores")