| from sklearn import preprocessing | |
| class FuncToNumber: | |
| def ToNumber(df): | |
| # transform non-numerical labels to numerical labels | |
| #Name,Sex,Age,SibSp,Parch,Ticket,Fare,Cabin,Embarked | |
| le = preprocessing.LabelEncoder() | |
| df["Sex"] = le.fit_transform(df["Sex"]) | |
| df["Age"] = le.fit_transform(df["Age"]) | |
| df["Ticket"] = le.fit_transform(df["Ticket"]) | |
| df["Fare"] = le.fit_transform(df["Fare"]) | |
| df["Cabin"] = le.fit_transform(df["Cabin"]) | |
| return df |