model_en_potion-base-8M Model Card

This Model2Vec model is a fine-tuned version of the potion-base-32M Model2Vec model. It also includes a classifier head on top.

Installation

Install model2vec using pip:

pip install model2vec[inference]

Usage

Load this model using the from_pretrained method:

from model2vec.inference import StaticModelPipeline

# Load a pretrained Model2Vec model
model = StaticModelPipeline.from_pretrained("model_en_potion-base-32M")

# Predict labels
predicted = model.predict(["Example sentence"])

Additional Resources

• Model2Vec Repo
• Model2Vec Base Models
• Model2Vec Results
• Model2Vec Tutorials
• Website

Library Authors

Model2Vec was developed by the Minish Lab team consisting of Stephan Tulkens and Thomas van Dongen.

Citation

Please cite the Model2Vec repository if you use this model in your work.

@article{minishlab2024model2vec,
  author = {Tulkens, Stephan and {van Dongen}, Thomas},
  title = {Model2Vec: Fast State-of-the-Art Static Embeddings},
  year = {2024},
  url = {https://github.com/MinishLab/model2vec}
}