From baa96e2a422198ac39c1653c4eca7f23647db8be Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?L=C3=A9o=20Bouscarrat?= <leo.bouscarrat@euranova.eu>
Date: Thu, 5 Mar 2020 10:59:45 +0100
Subject: [PATCH] Removing an unknown parameter for RaxModelResults
---
code/bolsonaro/trainer.py | 1 -
1 file changed, 1 deletion(-)
diff --git a/code/bolsonaro/trainer.py b/code/bolsonaro/trainer.py
index c543b85..aedf7f3 100644
--- a/code/bolsonaro/trainer.py
+++ b/code/bolsonaro/trainer.py
@@ -166,7 +166,6 @@ class Trainer(object):
if type(model) not in [RandomForestRegressor, RandomForestClassifier]:
results = ModelRawResults(
- model_object='',
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