{"metadata":{"kernelspec":{"language":"python","display_name":"Python 3","name":"python3"},"language_info":{"name":"python","version":"3.11.13","mimetype":"text/x-python","codemirror_mode":{"name":"ipython","version":3},"pygments_lexer":"ipython3","nbconvert_exporter":"python","file_extension":".py"},"kaggle":{"accelerator":"none","dataSources":[{"sourceId":93783,"databundleVersionId":11331308,"sourceType":"competition"},{"sourceId":525784,"sourceType":"modelInstanceVersion","isSourceIdPinned":true,"modelInstanceId":412049,"modelId":429842}],"dockerImageVersionId":31089,"isInternetEnabled":true,"language":"python","sourceType":"notebook","isGpuEnabled":false}},"nbformat_minor":4,"nbformat":4,"cells":[{"cell_type":"code","source":"%cd /\n!rm -rf /kaggle/working/esa_challenge\n!git clone https://github.com/lorenzoAllegrini/esa_challenge.git /kaggle/working/esa_challenge\n\n","metadata":{"_uuid":"8f2839f25d086af736a60e9eeb907d3b93b6e0e5","_cell_guid":"b1076dfc-b9ad-4769-8c92-a6c4dae69d19","trusted":true,"execution":{"iopub.status.busy":"2025-08-17T23:43:46.869154Z","iopub.execute_input":"2025-08-17T23:43:46.869422Z","iopub.status.idle":"2025-08-17T23:43:47.933523Z","shell.execute_reply.started":"2025-08-17T23:43:46.869396Z","shell.execute_reply":"2025-08-17T23:43:47.932506Z"}},"outputs":[],"execution_count":null},{"cell_type":"code","source":"!pip install --no-cache-dir --upgrade \"scikit-learn==1.5.1\" \"numpy==1.26.4\" \"scipy==1.11.4\"\n","metadata":{"trusted":true,"execution":{"iopub.status.busy":"2025-08-17T23:43:47.935539Z","iopub.execute_input":"2025-08-17T23:43:47.935833Z","iopub.status.idle":"2025-08-17T23:43:52.156662Z","shell.execute_reply.started":"2025-08-17T23:43:47.935804Z","shell.execute_reply":"2025-08-17T23:43:52.155475Z"}},"outputs":[],"execution_count":null},{"cell_type":"code","source":"!pip install Cython\n%cd /kaggle/working/esa_challenge/spaceai/segmentators\n!python setup.py build_ext --inplace","metadata":{"trusted":true,"execution":{"iopub.status.busy":"2025-08-17T23:43:52.157719Z","iopub.execute_input":"2025-08-17T23:43:52.157983Z","iopub.status.idle":"2025-08-17T23:44:03.0678Z","shell.execute_reply.started":"2025-08-17T23:43:52.15796Z","shell.execute_reply":"2025-08-17T23:44:03.066534Z"}},"outputs":[],"execution_count":null},{"cell_type":"code","source":"import sys\nsys.path.append('/kaggle/working/esa_challenge')\n\n\n\nfrom spaceai.segmentators.shapelet_miner import ShapeletMiner\nfrom spaceai.segmentators.esa_segmentator2 import EsaDatasetSegmentator2\n\nshapelet_miner = ShapeletMiner(\n    k_min_length=30, k_max_length=40, num_kernels=10,\n    segment_duration=50, step_duration=10,\n    run_id=\"esa_run\", exp_dir=\"experiments\", skip=True\n)\n\nsegmentator = EsaDatasetSegmentator2(\n    transformations=[\"min\",\"max\",\"mean\",\"std\",\"var\",\"stft\",\"sc\",\"slope\",\"diff_var\"],\n    segment_duration=50, step_duration=10,\n    shapelet_miner=shapelet_miner,\n    telecommands=False,\n    pooling_segment_len=200, pooling_segment_stride=20,\n    poolings=[\"max\",\"min\"],\n    run_id=\"esa_run\", exp_dir=\"experiments\",\n    use_shapelets=True\n)","metadata":{"trusted":true,"execution":{"iopub.status.busy":"2025-08-17T23:44:03.070086Z","iopub.execute_input":"2025-08-17T23:44:03.070934Z","iopub.status.idle":"2025-08-17T23:44:03.077006Z","shell.execute_reply.started":"2025-08-17T23:44:03.070899Z","shell.execute_reply":"2025-08-17T23:44:03.075968Z"}},"outputs":[],"execution_count":null},{"cell_type":"code","source":"from spaceai.benchmark.esa_competition_predictor import ESACompetitionPredictor\nfrom spaceai.utils.tools import kernel_column_selector\nimport os\n\n\npredictor = ESACompetitionPredictor(\"/kaggle/input/esa_predictor_artifacts/pytorch/default/3/inference_artifacts\", segmentator, data_root=\"/kaggle/input/esa_predictor_artifacts/pytorch/default/3/inference_artifacts\")\npredictor.predict(\"/kaggle/input/esa-adb-challenge/test.parquet\",\n                  \"/kaggle/working/submission.csv\")","metadata":{"trusted":true,"execution":{"iopub.status.busy":"2025-08-17T23:47:06.539528Z","iopub.execute_input":"2025-08-17T23:47:06.539803Z"}},"outputs":[],"execution_count":null}]}