Views
No views yet
XLNet-base-finetuned-HARPTxlnet, text-classification, privacy, trust, mobile-health, healthcare, harpt, custom-dataset, finetuned-modeldata_controldata_qualityrisksupportreliabilitycompetenceethicality1from transformers import XLNetForSequenceClassification, XLNetTokenizerFast
2
3# Load model and tokenizer
4model = XLNetForSequenceClassification.from_pretrained(
5 "tk648/XLNet-base-finetuned-HARPT",
6 use_safetensors=True
7)
8tokenizer = XLNetTokenizerFast.from_pretrained("tk648/XLNet-base-finetuned-HARPT")
9
10# Label mapping
11id2label = {
12 0: "competence",
13 1: "data control",
14 2: "data quality",
15 3: "ethicality",
16 4: "reliability",
17 5: "risk",
18 6: "support"
19}
20
21# Run prediction
22text = "This app crashes every time I open it."
23inputs = tokenizer(
24 text,
25 return_tensors="pt",
26 truncation=True,
27 max_length=512,
28 padding=True
29)
30outputs = model(**inputs)
31predicted_class_id = outputs.logits.argmax(dim=1).item()
32
33# Print predicted label
34predicted_label = id2label[predicted_class_id]
35print("Predicted label:", predicted_label)