Metadata-Version: 2.1
Name: NLP-AutoML
Version: 0.2
Summary: AutoML library for solving text -> label task
Home-page: UNKNOWN
Author: salaxieb
Author-email: salaxieb.ildar@gmail.com
License: UNKNOWN
Platform: UNKNOWN
Classifier: Programming Language :: Python :: 3
Classifier: License :: OSI Approved :: MIT License
Classifier: Operating System :: OS Independent
Description-Content-Type: text/markdown
Requires-Dist: pandas (==1.3.4)
Requires-Dist: scikit-learn (==1.0.1)
Requires-Dist: optuna (==2.10.0)
Requires-Dist: nltk (==3.6.5)
Requires-Dist: pymystem3 (==0.2.0)
Requires-Dist: fasttext (==0.9.2)
Provides-Extra: dev
Requires-Dist: tox (==3.24.4) ; extra == 'dev'
Requires-Dist: mypy (==0.910) ; extra == 'dev'
Requires-Dist: pylint (==2.11.1) ; extra == 'dev'
Requires-Dist: black (==21.10b0) ; extra == 'dev'
Requires-Dist: pytest (==6.2.5) ; extra == 'dev'

## AutoML NLP library
use to find baseline in text to label task

[Source code](https://github.com/salaxieb/NLP_AutoML)


#### Usage
```python
from sklearn.metrics import accuracy_score

from nlp_automl.auto_ml_pipeline import AutoMLPipeline

config = {
    'text_column': 'message',
    'target_column': 'intent',
    'dataset': dataset,
    'evaluator': accuracy_score,
}
automl = AutoMLPipeline(config=config)
best_params, pipeline = automl.find_solution(timeout=200)
preprocessor, vectorizer, model = pipeline
```
More detailed usage example [./examples/intent_prediction.py](https://github.com/salaxieb/NLP_AutoML/blob/master/examples/intent_predictition.py)


