Metadata-Version: 2.1
Name: airi-test-task
Version: 0.1.0
Summary: This library contains the code used to run a test job to AIRI.
License: Apache 2.0
Author: JohnConnor123
Author-email: ivan.eudokimoff2014@gmail.com
Maintainer: JohnConnor123
Maintainer-email: ivan.eudokimoff2014@gmail.com
Requires-Python: >=3.12,<3.13
Classifier: License :: Other/Proprietary License
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.12
Requires-Dist: autoviz (>=0.1.905,<0.2.0)
Requires-Dist: babyplots (>=1.7.0,<2.0.0)
Requires-Dist: catboost (>=1.2.5,<2.0.0)
Requires-Dist: chardet (>=5.2.0,<6.0.0)
Requires-Dist: charset-normalizer (>=3.3.2,<4.0.0)
Requires-Dist: dtale (>=3.13.1,<4.0.0)
Requires-Dist: flask-socketio (==5.2.0)
Requires-Dist: hydra-core (>=1.3.2,<2.0.0)
Requires-Dist: jupyter-bokeh (>=4.0.5,<5.0.0)
Requires-Dist: kaleido (>=0.2.1,<0.3.0,!=0.2.1.post1)
Requires-Dist: lightning (>=2.4.0,<3.0.0)
Requires-Dist: mlflow (>=2.16.0,<3.0.0)
Requires-Dist: numpy (<2.0.0)
Requires-Dist: optuna (>=4.0.0,<5.0.0)
Requires-Dist: optuna-dashboard (>=0.16.1,<0.17.0)
Requires-Dist: pandas (>=2.2.2,<3.0.0)
Requires-Dist: pca (>=2.0.7,<3.0.0)
Requires-Dist: plotly (==5.3.1)
Requires-Dist: pynvml (>=11.5.3,<12.0.0)
Requires-Dist: pytorch-lightning (==2.0.4)
Requires-Dist: ray[tune] (>=2.35.0,<3.0.0)
Requires-Dist: requests (>=2.32.3,<3.0.0)
Requires-Dist: rich (>=13.8.0,<14.0.0)
Requires-Dist: scikit-learn (>=1.5.1,<2.0.0)
Requires-Dist: sweetviz (>=2.3.1,<3.0.0)
Requires-Dist: tensorboard (>=2.17.1,<3.0.0)
Requires-Dist: torch-summary (>=1.4.5,<2.0.0)
Requires-Dist: tqdm (>=4.66.5,<5.0.0)
Requires-Dist: transformers (>=4.44.2,<5.0.0)
Requires-Dist: umap-learn (>=0.5.6,<0.6.0)
Requires-Dist: ydata-profiling (>=4.9.0,<5.0.0)
Requires-Dist: yellowbrick (>=1.5,<2.0)
Description-Content-Type: text/markdown

# Description
This library contains a complete development pipeline for performing an AIRI recruitment test task. 
Briefly, the goal of the problem is to predict the `DDG` value (change in protein stability due to mutations). 
A complete description of the test task can be found in the Jupyter notebook located at `src/notebooks/test_task.ipynb`.

# Installation:
### From GitHub
```
git clone https://github.com/JohnConnor123/airi-test-task.git
cd airi-test-task
poetry shell
cd src
source ./mlflow-init-commands.sh
python train.py
```
### From PyPI:
First, you need to download the `mlflow-init-commnads.sh` script in the `src` folder to start the mlflow server.
After that just do:
```
poetry add airi-test-task
```

