From 2bbf34ff43a3f43a5ce2ee2c69bfa3dff45b6af9 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ricardo=20Montan=CC=83ana?= <rmontanana@gmail.com>
Date: Fri, 1 Jan 2021 00:48:07 +0100
Subject: [PATCH] Fix kernel name in Models

---
 experimentation/Models.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/experimentation/Models.py b/experimentation/Models.py
index bb166d0..7e1433a 100644
--- a/experimentation/Models.py
+++ b/experimentation/Models.py
@@ -56,7 +56,7 @@ class ModelStree(ModelBase):
             "split_criteria": split_criteria,
             "max_features": max_features,
         }
-        self._poly = {
+        self._rbf = {
             "random_state": [self._random_state],
             "kernel": ["rbf"],
             "C": C,
@@ -65,7 +65,7 @@ class ModelStree(ModelBase):
             "split_criteria": split_criteria,
             "max_features": max_features,
         }
-        self._rbf = {
+        self._poly = {
             "random_state": [self._random_state],
             "kernel": ["poly"],
             "degree": [3, 5],