{"metadata":{"kernelspec":{"language":"python","display_name":"Python 3","name":"python3"},"language_info":{"name":"python","version":"3.10.12","mimetype":"text/x-python","codemirror_mode":{"name":"ipython","version":3},"pygments_lexer":"ipython3","nbconvert_exporter":"python","file_extension":".py"},"kaggle":{"accelerator":"none","dataSources":[{"sourceId":45867,"databundleVersionId":6924515,"sourceType":"competition"}],"dockerImageVersionId":30579,"isInternetEnabled":true,"language":"python","sourceType":"notebook","isGpuEnabled":false}},"nbformat_minor":4,"nbformat":4,"cells":[{"cell_type":"code","source":"import numpy as np\nimport pandas as pd\nfrom sklearn.model_selection import StratifiedKFold\nimport plotly.express as px","metadata":{"_uuid":"8f2839f25d086af736a60e9eeb907d3b93b6e0e5","_cell_guid":"b1076dfc-b9ad-4769-8c92-a6c4dae69d19","execution":{"iopub.status.busy":"2024-01-24T12:30:20.880615Z","iopub.execute_input":"2024-01-24T12:30:20.881566Z","iopub.status.idle":"2024-01-24T12:30:24.352702Z","shell.execute_reply.started":"2024-01-24T12:30:20.88148Z","shell.execute_reply":"2024-01-24T12:30:24.350689Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"train_df = pd.read_csv(\"/kaggle/input/UBC-OCEAN/train.csv\")","metadata":{"execution":{"iopub.status.busy":"2024-01-24T12:30:24.358801Z","iopub.execute_input":"2024-01-24T12:30:24.360437Z","iopub.status.idle":"2024-01-24T12:30:24.400115Z","shell.execute_reply.started":"2024-01-24T12:30:24.360346Z","shell.execute_reply":"2024-01-24T12:30:24.397935Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"label = train_df[\"label\"].values\nsplits = list(StratifiedKFold(n_splits=5, shuffle=True, random_state=42).split(X=np.zeros_like(label), y=label))","metadata":{"execution":{"iopub.status.busy":"2024-01-24T12:30:24.411974Z","iopub.execute_input":"2024-01-24T12:30:24.41367Z","iopub.status.idle":"2024-01-24T12:30:24.445503Z","shell.execute_reply.started":"2024-01-24T12:30:24.413587Z","shell.execute_reply":"2024-01-24T12:30:24.44268Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"for split_idx, (train_idx, val_idx) in enumerate(splits):\n    train_df[f\"split_{split_idx}\"] = train_df.index.isin(train_idx)","metadata":{"execution":{"iopub.status.busy":"2024-01-24T12:33:19.109601Z","iopub.execute_input":"2024-01-24T12:33:19.110099Z","iopub.status.idle":"2024-01-24T12:33:19.125769Z","shell.execute_reply.started":"2024-01-24T12:33:19.110066Z","shell.execute_reply":"2024-01-24T12:33:19.124423Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"px.histogram(data_frame=train_df, x=\"label\", facet_row=\"split_0\", color=\"is_tma\")","metadata":{"execution":{"iopub.status.busy":"2024-01-24T12:34:02.866064Z","iopub.execute_input":"2024-01-24T12:34:02.866449Z","iopub.status.idle":"2024-01-24T12:34:05.506012Z","shell.execute_reply.started":"2024-01-24T12:34:02.866418Z","shell.execute_reply":"2024-01-24T12:34:05.504884Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"train_df.to_csv(\"train_split.csv\")","metadata":{},"execution_count":null,"outputs":[]}]}