Skip to content
Snippets Groups Projects
Commit 8d834e32 authored by Leo Bouscarrat's avatar Leo Bouscarrat
Browse files

Update trainer.py

parent 95568681
Branches
No related tags found
1 merge request!18Resolve "Fix the save"
......@@ -166,7 +166,7 @@ class Trainer(object):
if type(model) not in [RandomForestRegressor, RandomForestClassifier]:
results = ModelRawResults(
model_object='',
model_weights='',
training_time=self._end_time - self._begin_time,
datetime=datetime.datetime.now(),
train_score=self.__score_func(model, self._dataset.X_train, self._dataset.y_train, False),
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment