How to use JayNightmare/PrERT-CNM with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="JayNightmare/PrERT-CNM")
# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("JayNightmare/PrERT-CNM") model = AutoModelForSequenceClassification.from_pretrained("JayNightmare/PrERT-CNM", device_map="auto")
cef688b
1
2
3
4
5
6
7
8
9
10
11
12
13
14
{ "labels": [ "user", "system", "organization" ], "model_name": "bert-base-uncased", "random_state": 42, "num_train_epochs": 2.0, "batch_size": 8, "learning_rate": 5e-05, "max_length": 256 }
