12 lines
308 B
Python
12 lines
308 B
Python
# 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()
|