Sentence Similarity
sentence-transformers
PyTorch
TensorFlow
Rust
ONNX
Safetensors
Transformers
English
bert
feature-extraction
Inference Endpoints
text-embeddings-inference
5 papers

Can I fine-tuning the model?

#56
by cathelloya - opened

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.

Sentence Transformers org

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

  • Tom Aarsen

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

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

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

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.

Sign up or log in to comment