refactor(Config): use a Config object instead of dictionary of dictionaries! (#184)

* refactor(Config): use a Config object instead of dictionary of dictionaries!

* fix(Config): use default_ensemble_config

* fix(Portfolio): fixed portfolio construction
This commit is contained in:
Mark Aron Szulyovszky
2022-01-23 18:37:43 +01:00
committed by GitHub
parent 5c4a5b0cf1
commit e80fffdb65
19 changed files with 223 additions and 196 deletions
+3 -3
View File
@@ -1,12 +1,12 @@
import pandas as pd
from config.config import Config
from utils.helpers import has_enough_samples_to_train
import warnings
def check_data(X:pd.DataFrame, y:pd.Series, training_config:dict):
def check_data(X:pd.DataFrame, y:pd.Series, config: Config):
""" Returns True if data is valid, else returns False."""
if has_enough_samples_to_train(X, y, training_config) == False:
if has_enough_samples_to_train(X, y, config) == False:
warnings.warn("Not enough samples to train")
return False