{"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":537370,"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","metadata":{"_uuid":"8f2839f25d086af736a60e9eeb907d3b93b6e0e5","_cell_guid":"b1076dfc-b9ad-4769-8c92-a6c4dae69d19","trusted":true},"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\"","metadata":{"trusted":true},"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},"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},"outputs":[],"execution_count":null},{"cell_type":"code","source":"test_file = \"/kaggle/input/esa-adb-challenge/test.parquet\"","metadata":{"trusted":true},"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/4/inference_artifacts\", segmentator, data_root=\"/kaggle/input/esa_predictor_artifacts/pytorch/default/4/inference_artifacts\")\nres = predictor.predict(test_file, \"/kaggle/working/submission.csv\")\nres.to_csv(\"/kaggle/working/submission.csv\", index=False)","metadata":{"trusted":true},"outputs":[],"execution_count":null}]}