gavindiaz/drift
1
0
Fork 0
mirror of https://github.com/webclinic017/drift.git synced 2026-08-04 06:37:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore(Linter): reformatted code with black (#211)

Browse Source 
* chore(Linter): reformatted code with black

* Create black.yaml
This commit is contained in:
Mark Aron Szulyovszky
2022-02-17 19:22:17 +01:00
committed by GitHub
parent f3fee4a4e1
commit 8dd2d88740
101 changed files with 2596 additions and 2320 deletions
Download Patch File Download Diff File Expand all files Collapse all files
data_loader/process.py
+10 -5
View File
@@ -4,16 +4,21 @@ import warnings
from utils.helpers import get_first_valid_return_index
from data_loader.types import XDataFrame
def check_data(X: XDataFrame, config: Config) -> bool:
""" Returns True if data is valid, else returns False."""
"""Returns True if data is valid, else returns False."""
if has_enough_samples_to_train(X, config) == False:
warnings.warn("Not enough samples to train")
return False
return True
def has_enough_samples_to_train(X: XDataFrame, config: Config) -> bool:
first_valid_index = get_first_valid_return_index(X.iloc[:,0])
first_valid_index = get_first_valid_return_index(X.iloc[:, 0])
samples_to_train = len(X) - first_valid_index
return samples_to_train > config.sliding_window_size_base + config.sliding_window_size_meta + 100
return (
samples_to_train
> config.sliding_window_size_base + config.sliding_window_size_meta + 100
)