TearingModeSurvival

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 the database

Training the model

Basic model analysis

Shap analysis

Name		Name	Last commit message	Last commit date
Latest commit History 38 Commits
README.md		README.md
auton_to_k2c.ipynb		auton_to_k2c.ipynb
auton_to_torch.ipynb		auton_to_torch.ipynb
data_processing.ipynb		data_processing.ipynb
data_processing_main.ipynb		data_processing_main.ipynb
format_survival_labels.py		format_survival_labels.py
get_survival_from_shot.py		get_survival_from_shot.py
hyperparam_model.cfg		hyperparam_model.cfg
hyperparameter_analyzer.ipynb		hyperparameter_analyzer.ipynb
hyperparameter_tuner.py		hyperparameter_tuner.py
job2.slurm		job2.slurm
launch_hyperparameter_tuning.py		launch_hyperparameter_tuning.py
launch_survival_training.py		launch_survival_training.py
metrics_helpers.py		metrics_helpers.py
model.cfg		model.cfg
paper_plotting_scripts.ipynb		paper_plotting_scripts.ipynb
plotting_helpers.py		plotting_helpers.py
shap_analysis.ipynb		shap_analysis.ipynb
shap_example.ipynb		shap_example.ipynb
tm_model_analysis.ipynb		tm_model_analysis.ipynb
tm_model_simple_analysis.ipynb		tm_model_simple_analysis.ipynb
tm_survival.ipynb		tm_survival.ipynb
train_survival_study.py		train_survival_study.py
train_tm_model.py		train_tm_model.py

Provide feedback