andre/100-Days-Of-ML-Code
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update Day 1_Data_Preprocessing.py

Browse Source
This commit is contained in:
yx-xyc
2021-01-14 10:59:16 +08:00
parent 524c84e8a9
commit 13020158fe
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
Code/Day 1_Data_Preprocessing.py
View File

@ -19,7 +19,7 @@ print(Y)
# from sklearn.impute import SimpleImputer
# imputer = SimpleImputer(missing_values="NaN", strategy="mean")
from sklearn.preprocessing import Imputer
imputer = Imputer(missing_values = "NaN", strategy = "mean", axis = 0)
imputer = Imputer(missing_values = np.nan, strategy = "mean", axis = 0)
imputer = imputer.fit(X[ : , 1:3])
X[ : , 1:3] = imputer.transform(X[ : , 1:3])
print("---------------------")
@ -33,7 +33,7 @@ from sklearn.preprocessing import LabelEncoder, OneHotEncoder
labelencoder_X = LabelEncoder()
X[ : , 0] = labelencoder_X.fit_transform(X[ : , 0])
#Creating a dummy variable
onehotencoder = OneHotEncoder(categorical_features = [0])
onehotencoder = OneHotEncoder('auto')
X = onehotencoder.fit_transform(X).toarray()
labelencoder_Y = LabelEncoder()
Y = labelencoder_Y.fit_transform(Y)