- Python 3.8.16
- 安裝相關package
pip install -r requirements.txt
- AI_CUP_medical.ipynb - 訓練病理模型
- AI_CUP_acoustic.ipynb - 訓練聲學模型
- AI_CUP_second-stage.ipynb - 訓練最終模型
- predict.ipynb - 利用上述三個檔案訓練好的模型進行資料預測
- 檔案執行順序: AI_CUP_medical.ipynb → AI_CUP_acoustic.ipynb → AI_CUP_second-stage.ipynb → predict.ipynb
- 說明:
- 執行各檔案前,必須先在前三個檔案中的「資料路徑設定」Cell下設定好病理資料與音訊資料的路徑以訓練模型,並在predict.ipynb中的「資料路徑設定」設定欲分類的資料集的路徑
- 訓練過程會自動產生儲存模型參數或路徑的檔案,請勿更動。包含以下檔案:
- AI_CUP_medical_model.joblib
- accoustic_model(folder)
- accoustic_model_path.txt
- final_model.joblib
https://drive.google.com/file/d/1i_BD31kJD0EuuyMv7q26inrYyVwx5yGx/view?usp=sharing