Model Card for Model ID
This model has been created with Argilla, trained with Spacy Explosion.
Model training
Training the model using the ArgillaTrainer
:
# Load the dataset:
dataset = FeedbackDataset.from_argilla("...")
# Create the training task:
def formatting_func(sample):
text = sample["text"]
label = sample["label"][0]["value"]
return(text, label)
task = TrainingTask.for_text_classification(formatting_func=formatting_func)
# Create the ArgillaTrainer:
trainer = ArgillaTrainer(
dataset=dataset,
task=task,
framework="spacy",
lang="en",
gpu_id=-1,
framework_kwargs={'optimize': 'efficiency', 'freeze_tok2vec': False},
)
trainer.train(output_dir="None")
You can test the type of predictions of this model like so:
trainer.predict("This is awesome!")
Model Details
Model Description
- Developed by: [More Information Needed]
- Shared by [optional]: [More Information Needed]
- Model type: [More Information Needed]
- Language(s) (NLP): [More Information Needed]
- License: [More Information Needed]
- Finetuned from model [optional]: [More Information Needed]
Technical Specifications [optional]
Framework Versions
- Python: 3.9.19
- Argilla: 1.28.0
- Downloads last month
- 0
Unable to determine this model’s pipeline type. Check the
docs
.