Skip to content

Machine Learning Transition State Analysis (MLTSA) suite with Analytical models to create data on demand and test the approach on different types of data and ML models.

License

Notifications You must be signed in to change notification settings

pedrojuanbj/MLTSA

Repository files navigation

MLTSA: Machine Learning Transition State Analysis repository

Introduction

This is a Python package to apply the MLTSA approach for relevant CV identification on Molecular Dynamics data using both Sklearn and TensorFlow modules.It also includes both a suite of 1D Potential Analytical model feature generation module for light testing and a suite of different 2D potential shapes (Spiral, Z-shaped) generation as well as the posterior feature generation by 1D projections of the 2D data. In this package you will find:

  • Data Generation Module (MLTSA_datasets) : Contains files with the easy to call 1D/2D/MD examples to generate data or play around with it as tests for the approach.
  • Scikit-Learn-based ML models and Feature Reduction module (MLTSA_sklearn) : Contains the Scikit-Learn integrated functions to apply MLTSA on data.
  • TensorFlow-based ML models and Feature Reduction module (MLTSA_tensorflow): Contains the set of functions and different models built on TensorFlow to apply MLTSA on data.

Usage

  • Example OneD
  • Example TwoD
  • Example Train
  • Example MLTSA

Installation

To use MLTSA, first install it using pip:

(.venv) $ pip install MLTSA