Skip to content
Snippets Groups Projects
Commit 1bfd15fe authored by Léo Bouscarrat's avatar Léo Bouscarrat
Browse files

Adding the missing argument

parent baa96e2a
No related branches found
No related tags found
1 merge request!15Resolve "Adding new datasets"
......@@ -166,6 +166,7 @@ class Trainer(object):
if type(model) not in [RandomForestRegressor, RandomForestClassifier]:
results = ModelRawResults(
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