Views
No views yet
1num_epochs: '2'
2model_name: distilbert-base-uncased
3learning_rate: 5e-05
4early_stopping_patience: None
5weight_decay: '0.01'
6batch_size: '32'
7max_length: '128'
8num_labels: '3'
91loss: 1.0714781284332275
2epoch: 2.0
31import mlflow
2
3# Load the model
4model = mlflow.pyfunc.load_model("runs:/ebe2ca3ecb634a96bf1ea3f65b2f86b9/intent_model")
5
6# Make predictions
7text = "your text here"
8prediction = model.predict([{"text": text}])