{"metadata":{"kernelspec":{"language":"python","display_name":"Python 3","name":"python3"},"language_info":{"pygments_lexer":"ipython3","nbconvert_exporter":"python","version":"3.6.4","file_extension":".py","codemirror_mode":{"name":"ipython","version":3},"name":"python","mimetype":"text/x-python"}},"nbformat_minor":4,"nbformat":4,"cells":[{"cell_type":"markdown","source":"## American Express: Create Training Pickle File","metadata":{}},{"cell_type":"code","source":"import numpy as np\nimport pandas as pd\nimport gc","metadata":{"execution":{"iopub.status.busy":"2022-05-26T17:55:03.067138Z","iopub.execute_input":"2022-05-26T17:55:03.068666Z","iopub.status.idle":"2022-05-26T17:55:03.075433Z","shell.execute_reply.started":"2022-05-26T17:55:03.068598Z","shell.execute_reply":"2022-05-26T17:55:03.074454Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"%%time\ntrain_data = pd.DataFrame()\nwith pd.read_csv('/kaggle/input/amex-default-prediction/train_data.csv', chunksize=10**5) as reader:\n    for counter, chunk in enumerate(reader):\n        for column in chunk.columns:\n            if str(chunk[column].dtype) == \"float64\":\n                chunk[column] = chunk[column].astype(np.float32)\n            if str(chunk[column].dtype) == \"int64\":\n                chunk[column] = chunk[column].astype(np.int32)\n        train_data = pd.concat([train_data, chunk])\n        gc.collect()","metadata":{"execution":{"iopub.status.busy":"2022-05-26T17:55:16.247650Z","iopub.execute_input":"2022-05-26T17:55:16.248094Z","iopub.status.idle":"2022-05-26T18:04:55.953245Z","shell.execute_reply.started":"2022-05-26T17:55:16.248060Z","shell.execute_reply":"2022-05-26T18:04:55.952436Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"train_data.to_pickle(\"train_data.pkl\")","metadata":{},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"train_labels = pd.read_csv(\"/kaggle/input/amex-default-prediction/train_labels.csv\")\ntrain_labels.to_pickle(\"train_labels.pkl\")","metadata":{},"execution_count":null,"outputs":[]}]}