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

Merge branch '18-fix-the-save' into 'master'

Resolve "Fix the save"

Closes #18

See merge request !18
parents 95568681 8d834e32
No related branches found
No related tags found
1 merge request!18Resolve "Fix the save"
...@@ -166,7 +166,7 @@ class Trainer(object): ...@@ -166,7 +166,7 @@ class Trainer(object):
if type(model) not in [RandomForestRegressor, RandomForestClassifier]: if type(model) not in [RandomForestRegressor, RandomForestClassifier]:
results = ModelRawResults( results = ModelRawResults(
model_object='', model_weights='',
training_time=self._end_time - self._begin_time, training_time=self._end_time - self._begin_time,
datetime=datetime.datetime.now(), datetime=datetime.datetime.now(),
train_score=self.__score_func(model, self._dataset.X_train, self._dataset.y_train, False), 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