README.md

Lexical Regressor v2

Ce modèle est basé sur CamemBERT, affiné pour effectuer de la régression lexicale sur 50 dimensions.

Exemple d'utilisation

from model import LexicalRegressor
from transformers import CamembertTokenizer
import torch

tokenizer = CamembertTokenizer.from_pretrained("jr303/lexical-regressor-MLP-2")
model = LexicalRegressor(output_dim=50)
model.load_state_dict(torch.load("pytorch_model.bin"))
model.eval()