Deep-learning-for-MS-inverse-design "test.py" file trains and validates ViT model "NN.py" file trains and validates MLP model "model_train.py" file trans and validates CNN model