Views
No views yet
distilbert-base-multilingual-cased that tags the technical issue a user complains
about in the free-text comment left when cancelling a cloud-gaming subscription. A single comment
may describe several problems at once, so this is a multi-label task over 6 classes.| label | meaning |
|---|---|
ping_latency | network delay, high ping, input lag |
frames_drop | low FPS, stuttering, unstable stream |
unable_launch | the game or service does not start |
mouse_keyboard_headset | peripherals: mouse, keyboard, headset, controller |
game_bug | a defect inside the game itself |
failed_save | progress is lost or not saved |
inference_config.json. They give
macro-F1 0.620 on the test set versus 0.581 with a plain 0.5 threshold.1import json
2
3import torch
4from huggingface_hub import hf_hub_download
5from transformers import AutoModelForSequenceClassification, AutoTokenizer
6
7REPO = "dar1bi/cloud-gaming-feedback-multilabel"
8
9tokenizer = AutoTokenizer.from_pretrained(REPO)
10model = AutoModelForSequenceClassification.from_pretrained(REPO).eval()
11config = json.load(open(hf_hub_download(REPO, "inference_config.json")))
12
13text = "Constant micro stutter and very low fps in The Finals"
14encoded = tokenizer(text, truncation=True, padding="max_length",
15 max_length=config["max_len"], return_tensors="pt")
16with torch.no_grad():
17 probabilities = torch.sigmoid(model(**encoded).logits).numpy()[0]
18
19labels = [label for label, probability, threshold
20 in zip(config["labels"], probabilities, config["thresholds"])
21 if probability >= threshold]
22print(labels) # ['frames_drop']distilbert-base-multilingual-casedBCEWithLogitsLoss with pos_weight to compensate for class imbalance| split | macro-F1 | micro-F1 | macro ROC-AUC | mAP |
|---|---|---|---|---|
| validation | 0.681 | 0.717 | 0.828 | 0.727 |
| test | 0.620 | 0.678 | 0.817 | 0.685 |
ping_latency 0.78, unable_launch 0.70,
mouse_keyboard_headset 0.67, frames_drop 0.62, game_bug 0.49, failed_save 0.46.game_bug and failed_save are recognised in fewer than half of real cases —
they are infrequent in the data and semantically overlap with the other classes.