{"cells":[{"metadata":{"_uuid":"ebb70d7332fcafed13c1c963652c48285b531b76","_cell_guid":"504c1194-6980-45aa-92f1-caf55b44dbb0"},"cell_type":"markdown","source":"# 九章 KNN 教程 \n\nhttps://www.kaggle.com/c/digit-recognizer"},{"metadata":{"_uuid":"26283408c3c54401ae8c237dded11a5d54141159","_cell_guid":"0604f6b3-046f-4b65-a1a9-3a5f8e824752"},"cell_type":"markdown","source":"# 导入数据"},{"metadata":{"_uuid":"11ab637213812ce08688490c3f6cef097d603ff2","_cell_guid":"f206c2ba-7c09-4f26-ac95-e8a24dfb9ae7","trusted":true},"cell_type":"code","source":"import numpy as np # 数组常用库\nimport pandas as pd # 读入csv常用库\nimport matplotlib.pyplot as plt # 画图常用库","execution_count":null,"outputs":[]},{"metadata":{"_uuid":"ae5b578dffd73ed55fcba8b4c011334d46f55027","_cell_guid":"61d6082e-6ea1-4f92-9329-0848647a17c3"},"cell_type":"markdown","source":"从 `../train.csv` and `../test.csv` 文件里面读入 numpy arrays.\n\n导入数据后打印shape查看是否数据是否完整\n\n导入数据差不多10秒"},{"metadata":{"_uuid":"8541a8ca7e77c1b6885d56f8b7b932b13f5f67c8","_cell_guid":"cc5d4fe8-a0a6-4640-973c-94bfe7639635","trusted":true},"cell_type":"code","source":"def load_data(data_dir, train_row):\n    train = pd.read_csv(data_dir + \"train.csv\")\n    X_train = train.values[:train_row, 1:]\n    y_train = train.values[:train_row, 0]\n    Pred_test = pd.read_csv(data_dir + \"test.csv\")\n    return X_train, y_train, Pred_test\n\nOrigin_X_train, Origin_y_train, Origin_test = load_data(\"../input/\", 5000)","execution_count":null,"outputs":[]},{"metadata":{"_uuid":"df3d5ab5655dc2f05367d6c147f7423effcf1d68","_cell_guid":"4c2b43d7-5930-4e98-af99-d74049771acc"},"cell_type":"markdown","source":"# 用图像展示一下数据集"},{"metadata":{"_uuid":"4eb06a693881cc878ccd9fc9647af48b2743b0ce","_cell_guid":"5c654f1a-4c8c-4b7d-b1e8-5fae2ff048c5"},"cell_type":"markdown","source":"\n展示部分数据看看图片是什么样子的。（有兴趣的同学可以自己看）"},{"metadata":{"_uuid":"761de08af4ee6c975fe406ed66f90f0bc88fe8ff","_cell_guid":"c2c6ec59-ebc1-4d4e-ae89-d5a9ef240835","trusted":true},"cell_type":"code","source":"row = 5\nprint(Origin_y_train[row])\nplt.imshow(Origin_X_train[row].reshape((28, 28)))\nplt.show()","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"rows = 4\nclass_num = 10\nfor num in range(class_num):\n    idxs = np.nonzero([i == num for i in Origin_y_train])\n    idxs = np.random.choice(idxs[0], rows)\n    for i, idx in enumerate(idxs):\n        plt_idx = i * class_num + num + 1\n        plt.subplot(rows, class_num, plt_idx)\n        plt.imshow(Origin_X_train[idx].reshape((28, 28)))\n        plt.axis(\"off\")\n        if i == 0:\n            plt.title(num)\nplt.show()","execution_count":null,"outputs":[]},{"metadata":{"_uuid":"31a53b52bea59d427d33194aafd6103ad46cf601","_cell_guid":"16e64324-48af-492d-8f42-c6ac6e7a35c0"},"cell_type":"markdown","source":"把原本的训练数据拆分成为20%测试集（8400个），80%训练集（33600个）\n\n怎么样2/8分"},{"metadata":{"_uuid":"9f3273d2f90dc599f82c63f3bc829b779e282eb5","_cell_guid":"ff788363-fa17-4748-bba8-6a4ae2487181","trusted":true},"cell_type":"code","source":"from sklearn.model_selection import train_test_split\nX_train, X_vali, y_train, y_vali = train_test_split(Origin_X_train,\n                                                   Origin_y_train,\n                                                   test_size = 0.2,\n                                                   random_state = 0)\nprint(X_train.shape, X_vali.shape, y_train.shape, y_vali.shape)","execution_count":null,"outputs":[]},{"metadata":{"_uuid":"d1d5b8c1013bbb069b93fc79791ddaa2d1607277","_cell_guid":"68bc3069-1241-4dc1-a28c-46ca0d22da5d"},"cell_type":"markdown","source":"\n下面进行KNN模型训练\n如果是大数据，需要1.5小时跑完"},{"metadata":{"_uuid":"2d8a667aa10291fe03b528c4ab5f55eb7dcf9dd0","_cell_guid":"93aec800-e499-4de7-a218-9d3231043b32","trusted":true},"cell_type":"code","source":"import time\nfrom sklearn.metrics import accuracy_score, classification_report, confusion_matrix\nfrom sklearn.neighbors import KNeighborsClassifier\n\nk_range = range(1, 10)\nscores = []\n\nfor k in k_range:\n    print(\"k = \" + str(k) + \" begin \")\n    start = time.time()\n    knn = KNeighborsClassifier(n_neighbors=k)\n    knn.fit(X_train, y_train)\n    y_pred = knn.predict(X_vali)\n    acc = accuracy_score(y_vali, y_pred)\n    scores.append(acc)\n    end = time.time()\n    print(classification_report(y_vali, y_pred))\n    print(confusion_matrix(y_vali, y_pred))\n    print(\"Complete time:\" + str(start - end) + \" Secs.\")","execution_count":null,"outputs":[]},{"metadata":{"_uuid":"e11013543394219528ba34e266c9134e9b340ebf","_cell_guid":"85741aa6-14e1-43e3-b12c-f00180ea9db8"},"cell_type":"markdown","source":"# 通过训练我们可以找到最好的测试数据集，打印出Accuracy的图"},{"metadata":{"trusted":true},"cell_type":"code","source":"print (scores)\nplt.plot(k_range,scores)\nplt.xlabel('Value of K')\nplt.ylabel('Testing accuracy')\nplt.show()","execution_count":null,"outputs":[]},{"metadata":{"_uuid":"fdbf617c65d78718fe7793e7fc2e55ecadaaa04e","_cell_guid":"0fc75b9d-87e5-4147-b74f-724d7bc93fcd"},"cell_type":"markdown","source":"# 预测测试数据集的结果\n"},{"metadata":{"_uuid":"1d791d62060f3b6deb5215c89eb6e813b95bd536","_cell_guid":"87c41cfe-4863-4e7f-a38b-dabdf14b24a0","trusted":true},"cell_type":"code","source":"k = 3\n\nknn = KNeighborsClassifier(n_neighbors=k)\nknn.fit(Origin_X_train,Origin_y_train)\ny_pred = knn.predict(Origin_test[:300])","execution_count":null,"outputs":[]},{"metadata":{"_uuid":"6890a00551cdf3ec76ee99a60965a670ed20c239","_cell_guid":"02616849-0f6c-422f-82b1-f1b7a31c87c5"},"cell_type":"markdown","source":"# 验证一下结果"},{"metadata":{"_uuid":"f5456e57ecb856612c2be0e7b3089a5e4a34dd5a","_cell_guid":"6d774516-3d31-417a-b1eb-d7a8ffa6242e","trusted":true},"cell_type":"code","source":"print (y_pred[200])\nplt.imshow(Origin_test.values[200].reshape((28, 28)))\nplt.show()","execution_count":null,"outputs":[]},{"metadata":{"_uuid":"401c578da3e3c6ea4180242bc5843d7a8cb00b65","_cell_guid":"eadc71f0-386e-4318-8261-d3a0da05c0bd"},"cell_type":"markdown","source":"# 预测之后，我们把预测好的数据打印到文件Digit_Recogniser_Result.csv里面。\n# 打印的格式是如下\n\n* ImageId\tLabel\n* 1\t2\n* 2\t0\n* 3\t9\n* 4\t9"},{"metadata":{"_uuid":"6b467e130ce5d8f5656062c7eadf63621d171671","_cell_guid":"a7c7dd6d-6d89-4ea9-9d37-41a6d3ecf048","trusted":true},"cell_type":"code","source":"print(len(y_pred))\n\npd.DataFrame({\"ImageId\": list(range(1,len(y_pred)+1)),\"Label\": y_pred}).to_csv('Digit_Recogniser_Result.csv', index=False,header=True)","execution_count":null,"outputs":[]}],"metadata":{"kernelspec":{"display_name":"Python 3","language":"python","name":"python3"},"language_info":{"name":"python","version":"3.6.6","mimetype":"text/x-python","codemirror_mode":{"name":"ipython","version":3},"pygments_lexer":"ipython3","nbconvert_exporter":"python","file_extension":".py"}},"nbformat":4,"nbformat_minor":4}