{"metadata":{"kernelspec":{"language":"python","display_name":"Python 3","name":"python3"},"language_info":{"name":"python","version":"3.10.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":71549,"databundleVersionId":8561470,"sourceType":"competition"}],"dockerImageVersionId":30746,"isInternetEnabled":false,"language":"python","sourceType":"notebook","isGpuEnabled":false}},"nbformat_minor":4,"nbformat":4,"cells":[{"cell_type":"code","source":"import pandas as pd\n\ndf = pd.read_csv(\n    '/kaggle/input/rsna-2024-lumbar-spine-degenerative-classification/test_series_descriptions.csv'\n)\n\n# need to predict every condiction_level (total 25) for every study\nCONDITIONS = [\n    'spinal_canal_stenosis', \n    'left_neural_foraminal_narrowing', \n    'right_neural_foraminal_narrowing',\n    'left_subarticular_stenosis',\n    'right_subarticular_stenosis'\n]\nLEVELS = [\n    'l1_l2',\n    'l2_l3',\n    'l3_l4',\n    'l4_l5',\n    'l5_s1',\n]\n\nfrom random import random\nimport numpy as np\n\ndef predict():\n    # return [\"normal_mild\", \"moderate\", \"severe\"] scores for every studyid_condition_level\n    return [0.6, 0.3, 0.1]\n\nsubmission_df = []\nfor study_id in df[\"study_id\"].unique():\n    for condition in CONDITIONS:\n        for level in LEVELS:\n            new_row = [\n                \"_\".join([str(study_id), condition, level])\n            ]\n            new_row += predict()\n            submission_df.append(new_row)\nsubmission_df = pd.DataFrame(submission_df, columns=[\"row_id\", \"normal_mild\", \"moderate\", \"severe\"])\nsubmission_df.to_csv('submission.csv', index=False)","metadata":{"execution":{"iopub.status.busy":"2024-08-07T03:16:45.207910Z","iopub.execute_input":"2024-08-07T03:16:45.208315Z","iopub.status.idle":"2024-08-07T03:16:46.351821Z","shell.execute_reply.started":"2024-08-07T03:16:45.208282Z","shell.execute_reply":"2024-08-07T03:16:46.350768Z"},"trusted":true},"execution_count":null,"outputs":[]}]}