Hugging Face's logo

sentence-transformers
/
all-MiniLM-L6-v2

Sentence Similarity
sentence-transformers
PyTorch
TensorFlow
Rust
ONNX
Safetensors
OpenVINO
Transformers
English
bert
feature-extraction
Eval Results
text-embeddings-inference
Model card Files
xet
 Community
157

Can I fine-tuning the model?

#56
by cathelloya - opened
Discussion
cathelloya

Can I add my own dataset to fine-tuning the model? I want to add some data in professional field, maybe it can work better.

tomaarsen
 Sentence Transformers org

Yes, you can. There's some training examples here: https://sbert.net/examples/training/sts/README.html

  • Tom Aarsen
cathelloya

Yes, you can. There's some training examples here: https://sbert.net/examples/training/sts/README.html

  • Tom Aarsen

Thanks for your reply. :)
cathelloya changed discussion status to closed
secsilm

@tomaarsen Is it suitable for regression or classification tasks? Is it suitable for these tasks?
tomaarsen
 Sentence Transformers org

I've not experimented with regression, but I think it should be suitable, yes.
For classification, certainly yes. We've developed SetFit for training classification models using Sentence Transformer models. Here's an overview of how & why SetFit works: https://huggingface.co/docs/setfit/conceptual_guides/setfit

  • Tom Aarsen
secsilm

I've not experimented with regression, but I think it should be suitable, yes.
For classification, certainly yes. We've developed SetFit for training classification models using Sentence Transformer models. Here's an overview of how & why SetFit works: https://huggingface.co/docs/setfit/conceptual_guides/setfit

  • Tom Aarsen

Thanks.
mimizai

very helpful.

Sign up or log in to comment