refactor(Project): removed regression method (we can still use regression models, but we'll need map them to classification later) (#182)

* refactor(Project): removed regression method (we can still use regression models, but we'll need map them to classes later)

* fix(Training): removed mistakenly left in `method` parameter
This commit is contained in:
Mark Aron Szulyovszky
2022-01-23 17:15:08 +01:00
committed by GitHub
parent 516c8bcc87
commit 5c4a5b0cf1
12 changed files with 11 additions and 44 deletions
-3
View File
@@ -2,7 +2,6 @@ from utils.evaluate import discretize_threeway_threshold, evaluate_predictions
from utils.helpers import equal_except_nan
from training.primary_model import train_primary_model
import pandas as pd
from models.model_map import default_feature_selector_classification, default_feature_selector_regression
from models.base import Model
from reporting.types import Reporting
from typing import Union, Optional
@@ -35,7 +34,6 @@ def train_meta_labeling_model(
y = meta_y,
target_returns = target_returns,
models = models,
method = 'classification',
expanding_window = training_config['expanding_window_meta_labeling'],
sliding_window_size = training_config['sliding_window_size_meta_labeling'],
retrain_every = training_config['retrain_every'],
@@ -59,7 +57,6 @@ def train_meta_labeling_model(
target_returns = target_returns,
y_pred = avg_predictions_with_sizing,
y_true = y,
method = 'classification',
no_of_classes = 'two',
print_results = True,
discretize=False