A tearing mode prediction model based on the auton-survival deep survival machine.
For simply trying out the model:
- The models, data and configs are located in the Princeton University clusters such as stellar, in /projects/EKOLEMEN/survival_tm_2/ in folders of their name
- Use tm_model_simple_analysis.ipynb for trying out predictions on any shot in the database
For more advanced use, the repository includes the following features:
- Creating and formatting the database from the data-fetching repo
- Training the model
- Basic model analysis and inference
- Shap analysis
Creating the database
Training the model
Basic model analysis
Shap analysis