mirror of
https://github.com/webclinic017/drift.git
synced 2026-07-28 03:08:01 +00:00
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:
committed by
GitHub
parent
5c4a5b0cf1
commit
e80fffdb65
@@ -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
|
||||
|
||||
|
||||
Reference in New Issue
Block a user