Skip to content

Machine learning and artificial intelligence benchmarking library for nuclear engineering applications.

License

Notifications You must be signed in to change notification settings

aims-umich/pyMAISE

Repository files navigation

pyMAISE: Michigan Artificial Intelligence Standard Environment

License Tests Status Documentation Status

pyMAISE is an artificial intelligence (AI) and machine learning (ML) benchmarking library for nuclear reactor applications. It offers to streamline the building, tuning, and comparison 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.

These models are built using scikit-learn and Keras. 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.

Installation and Documentation

Refer to the installation guide and documentation for help.

Benchmark Jupyter Notebooks

You can find the pyMAISE benchmarks here or below.

About

Machine learning and artificial intelligence benchmarking library for nuclear engineering applications.

Resources

License

Stars

Watchers

Forks

Packages

No packages published