{"cells":[{"metadata":{"_uuid":"26283408c3c54401ae8c237dded11a5d54141159","_cell_guid":"0604f6b3-046f-4b65-a1a9-3a5f8e824752","trusted":true},"cell_type":"code","source":"# 导入数据","execution_count":null,"outputs":[]},{"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":{"trusted":true},"cell_type":"code","source":"data_dir = \"../input/\"\n\ndef load_data(data_dir, train_row):\n    train = pd.read_csv(data_dir + \"train.csv\")\n    print(train.shape)\n    \n    X_train = train.values[0:train_row, 1:]\n    y_train = train.values[0:train_row, 0]\n    \n    Pred_test = pd.read_csv(data_dir + \"test.csv\").values\n    \n    return X_train, y_train, Pred_test","execution_count":null,"outputs":[]},{"metadata":{"_uuid":"8541a8ca7e77c1b6885d56f8b7b932b13f5f67c8","_cell_guid":"cc5d4fe8-a0a6-4640-973c-94bfe7639635","trusted":true},"cell_type":"code","source":"train_row = 5000\nOrigin_X_train, Origin_y_train, Origin_X_test = load_data(data_dir, train_row)","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"print(Origin_X_train.shape, Origin_y_train.shape, Origin_X_test.shape)\nprint(Origin_X_train)","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":{"trusted":true},"cell_type":"code","source":"import matplotlib\nimport matplotlib.pyplot as plt\n\nrow = 3\n\nprint (Origin_y_train[row])\n\nplt.imshow(Origin_X_train[row].reshape((28, 28)))\nplt.show()\n\nclasses = [\"0\", \"1\", \"2\", \"3\", \"4\", \"5\", \"6\", \"7\", \"8\", \"9\"]\nrow = 4\n\nprint(classes)\nfor y, cls in enumerate(classes):\n    idxs = np.nonzero([i==y for i in Origin_y_train])\n    idxs = np.random.choice(idxs[0], row)\n    for i, idx in enumerate(idxs):\n        plt_idx = i*len(classes) + y + 1\n        plt.subplot(row, len(classes), plt_idx)\n        plt.imshow(Origin_X_train[idx].reshape((28, 28)))\n        plt.axis(\"off\")\n        \n        if i == 0:\n            plt.title(cls)\nplt.show()\n\n","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\n\nX_train, X_val, y_train, y_val = train_test_split(Origin_X_train, Origin_y_train, test_size = 0.2, random_state = 0)\n\nprint(X_train.shape, X_val.shape, y_train.shape, y_val.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\nans_k = 0\n\nscores = []\n\nfor k in range(1, 8):\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_val)\n    accuracy = accuracy_score(y_val, y_pred)\n    scores.append(accuracy)\n    end = time.time()\n    \n    print(classification_report(y_val, y_pred))\n    print(confusion_matrix(y_val, y_pred))\n    \n    print(\"Complete time: \" + str(end - start) + \"sec\")\n","execution_count":null,"outputs":[]},{"metadata":{"_uuid":"e11013543394219528ba34e266c9134e9b340ebf","_cell_guid":"85741aa6-14e1-43e3-b12c-f00180ea9db8"},"cell_type":"markdown","source":"# 通过训练我们可以找到最好的测试数据集，打印出Accuracy的图"},{"metadata":{"_uuid":"1862c48c0d9ccba622f7876d5baad0191bf32796","_cell_guid":"67390da3-6710-4bce-b0f1-8f364e5944e1","trusted":true},"cell_type":"code","source":"plt.plot(range(1, 8), scores)\nplt.xlabel(\"k Value\")\nplt.ylabel(\"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 = 4\n    knn = KNeighborsClassifier(n_neighbors = k)\n    knn.fit(Origin_X_train, Origin_y_train)\n    y_pred = knn.predict(Origin_X_test[:500])","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[300])\nplt.imshow(Origin_X_test[300].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":"pd.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":1}