import?pandas?as?pd
from?sklearn.model_selection?import?train_test_split
from?sklearn.feature_extraction?import?DictVectorizer
from?sklearn.tree?import?DecisionTreeClassifier?export_graphviz
from?sklearn.model_selection?import?RandomizedSearchCV


def?tree（）:
????#?（1）獲取數據
????titanic?=?pd.read_csv（“tree_titanic.txt“）
????#?“row.names““pclass““survived““name““age““embarked““home.dest““room““ticket““boat““sex“
????print（titanic.head（））

????#?（2）數據處理?-?需要進行挑選特征值和目標值
????x?=?titanic[[“pclass“?“age“?“sex“]]
????y?=?titanic[“survived“]
????print（“特征：\n“?x.head（））
????#?（2）數據處理?-?因為年齡數據有缺失，則需要進行填充：采用年齡的平均值進行補充
????x[“age“].fillna（x[“age“].mean（）?inplace=True）
????#?（2）數據處理?-?轉換成字典，然后進行特征工程的字典抽取
????x?=?x.to_dict（orient=“records“）
????print（“轉換成字典之后的特征：\n“?x）

????#?（3）劃分訓練集、測試集
????x_train?x_test?y_train?y_test?=?train_test_split（x?y?random_state=22）

????#?（4）特征工程-進行字典特征抽取
????transfer?=?DictVectorizer（）
????x_train?=?transfer.fit_transform（x_train）
????x_test?=?transfer.transform（x_test）

????print（“特征工程之后的訓練集的特征：\n“?x_train）
????print（“特征：\n“?transfer.get_feature_names（））

????#（5）訓練模型

????classifier?=?DecisionTreeClassifier（）
????param?=?{“max_depth“:?range（250?1）}
????#classifier?=?RandomizedSearchCV（classifier?param_distributions=param?cv=8）
????classifier.fit（x_train?y_train）

????y_predict?=?classifier.predict（x_test）
????print（“測試結果為：\n“?y_predict?==?y_test）
????#??print（“最好預估器為：\n“?classifier.best_estimator_）

????#?（6）模型評估
????score?=?classifier.score（x_testy_test）
????print（“準確度為：?“?score）

????#?（7）可視化
????export_graphviz（classifier?out_file=“/Users/j1/Documents/機器學習/code/machinelearning/estimator/tree_titanic.dot“?feature_names=transfer.get_feature_names（））

????return?None


if?__name__?==?‘__main__‘:
????tree（）

?屬性????????????大小?????日期????時間???名稱
-----------?---------??----------?-----??----
?????目錄???????????0??2019-12-04?13:13??tree_titanic\
?????文件??????734945??2019-12-04?10:22??tree_titanic\tree_titanic.png
?????文件????????2149??2019-12-04?13:13??tree_titanic\tree_titanic.py
?????目錄???????????0??2019-12-04?13:14??__MACOSX\
?????目錄???????????0??2019-12-04?13:14??__MACOSX\tree_titanic\
?????文件?????????176??2019-12-04?13:13??__MACOSX\tree_titanic\._tree_titanic.py
?????文件??????219624??2019-12-04?11:50??tree_titanic\tree_titanic1.png
?????文件??????115631??2019-12-03?14:38??tree_titanic\tree_titanic.txt
?????文件?????????233??2019-12-03?14:38??__MACOSX\tree_titanic\._tree_titanic.txt