AutoML library for solving text -> label task
Project description
AutoML NLP library
use to find baseline in text to label task
Usage
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
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
NLP AutoML-0.2.tar.gz
(7.9 kB
view hashes)
Built Distribution
NLP_AutoML-0.2-py3-none-any.whl
(12.9 kB
view hashes)