polymer_viscosity_prediction

Follow the commands below on terminal for reproduction.

Installation

git clone https://github.com/hhaootian/polymer_viscosity_prediction
cd polymer_viscosity_prediction
pip install -r requirements.txt
cd src

Model Training

python src/fine_tune.py
python src/test.py

Plotting

python src/plot_performance.py
python src/plot_importance.py

License

GPL v3