Intent Classification Model

This is an intent classification model trained using MLflow and uploaded to the Hugging Face Hub.

Model Details

  • Model Type: Intent Classification
  • Framework: MLflow
  • Run ID: ebe2ca3ecb634a96bf1ea3f65b2f86b9

Training Details

Parameters

num_epochs: '2'
model_name: distilbert-base-uncased
learning_rate: 5e-05
early_stopping_patience: None
weight_decay: '0.01'
batch_size: '32'
max_length: '128'
num_labels: '3'

Metrics

loss: 1.0714781284332275
epoch: 2.0

Usage

This model can be used to classify intents in text. It was trained using MLflow and can be loaded using the MLflow model registry.

Loading the Model

import mlflow

# Load the model
model = mlflow.pyfunc.load_model("runs:/ebe2ca3ecb634a96bf1ea3f65b2f86b9/intent_model")

# Make predictions
text = "your text here"
prediction = model.predict([{"text": text}])

Additional Information

For more information about using this model or the training process, please refer to the repository documentation.

Downloads last month

-

Downloads are not tracked for this model. How to track
Inference Providers NEW
This model is not currently available via any of the supported Inference Providers.
The model cannot be deployed to the HF Inference API: The HF Inference API does not support text-classification models for mlflow library.