pyMAISE: Michigan Artificial Intelligence Standard Environment

pyMAISE is an artificial intelligence (AI) and machine learning (ML) benchmarking library for nuclear reactor applications. It offers to streamline the building, tuning, comparison, and explainability of various ML models for user-provided data sets. Also, pyMAISE offers benchmarked data sets, written in Jupyter Notebooks, for AI/ML comparison. Current ML algorithm support includes

• linear regression,
• lasso regression,
• logistic regression,
• decision tree regression and classification,
• support vector regression and classification,
• random forest regression and classification,
• k-nearest neighbors regression and classification,
• sequential neural networks.

Additionally, pyMAISE supports basic explainability analysis via SHAP for all the ML algorithms listed above. An example is provided in the CHF benchmark. Current SHAP support includes

• DeepLIFT,
• Integrated Gradients,
• Kernel SHAP,
• and Exact SHAP.

These models are built using scikit-learn and Keras with explainability using SHAP. pyMAISE supports the following neural network layers:

• dense,
• dropout,
• LSTM,
• GRU,
• 1D, 2D, and 3D convolutional,
• 1D, 2D, and 3D max pooling,
• flatten,
• and reshape.

Please use the following to reference pyMAISE:

@article{MYERS2025105568,
    title = {pyMAISE: A Python platform for automatic machine learning
        and accelerated development for nuclear power applications},
    journal = {Progress in Nuclear Energy},
    volume = {180},
    pages = {105568},
    year = {2025},
    issn = {0149-1970},
    doi = {https://doi.org/10.1016/j.pnucene.2024.105568},
    url = {
        https://www.sciencedirect.com/science/article/pii/S0149197024005183
    },
    author = {Patrick A. Myers and Nataly Panczyk and Shashank Chidige
        and Connor Craig and Jacob Cooper and Veda Joynt and Majdi
        I. Radaideh},
}

Installation and Documentation

Refer to the installation guide and documentation for help.

Benchmark Jupyter Notebooks

You can find the pyMAISE benchmarks here or below.

• MIT Reactor
• Reactor Physics
• Fuel Performance
• Heat Conduction
• BWR Micro Core
• HTGR Micro-Core Quadrant Power
• NEACRP C1 Rod Ejection Accident
• Critical Heat Flux (CHF) Prediction
• Binary Anomaly Detection