diff --git a/py_scripts/scout.py b/py_scripts/scout.py index 621de4c..424e0a8 100644 --- a/py_scripts/scout.py +++ b/py_scripts/scout.py @@ -1 +1,11 @@ # Regression/Prediction (Totally gonna do later trust bro) + +from sklearn.impute import SimpleImputer +import numpy as np + + +def cure_depression(dataset, column): + dataset[column].fillna(dataset[column].mean(), inplace=True) + print("what you want") + print(dataset) + return dataset[column].to_list()