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 import AutoMLPipeline
task = {
'text_column': 'description', # required
'target_column': 'user_type', # required
'dataset': dataset, # required
'use_label_encoder': True, # Optional, default: True
'evaluator': accuracy_score, # required
'fit_pipeline': True, # Optional, default: True
}
automl = AutoMLPipeline()
best_params, pipeline = automl.find_solution(task=task, 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.3.tar.gz
(8.7 kB
view hashes)
Built Distribution
NLP_AutoML-0.3-py3-none-any.whl
(14.6 kB
view hashes)