From 1bfd15feee13aae0b525a6bf081a91ac47834494 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?L=C3=A9o=20Bouscarrat?= <leo.bouscarrat@euranova.eu>
Date: Thu, 5 Mar 2020 11:00:53 +0100
Subject: [PATCH] Adding the missing argument
---
code/bolsonaro/trainer.py | 1 +
1 file changed, 1 insertion(+)
diff --git a/code/bolsonaro/trainer.py b/code/bolsonaro/trainer.py
index aedf7f3..0e437d2 100644
--- a/code/bolsonaro/trainer.py
+++ b/code/bolsonaro/trainer.py
@@ -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),
--
GitLab