joshswift/phobihsd-proposed
PhoBiHSD proposed checkpoint (PhoBERT-BiLSTM) for Vietnamese hate speech detection on ViHSD.
Files
phobihsd_proposed.pt: PyTorch checkpoint (state_dict+ optional thresholds).phobihsd_proposed.meta.json: model metadata.model_comparison.yaml: training/inference config used by this repository.
Label Mapping
0: Clean1: Offensive2: Hate
Metadata (from checkpoint sidecar)
{
"model_name": "vinai/phobert-base-v2",
"head_type": "cls_mlp",
"hidden_dim": 256,
"dropout": 0.5,
"freeze_encoder": false,
"max_len": 100,
"thresholds": [
0.4,
0.4,
0.6
],
"seed": 4
}
Inference (from this repo)
export PHOBIHSD_PROPOSED_CKPT=phobihsd_proposed.pt
export PHOBIHSD_CONFIG=config/experiments/model_comparison.yaml
python -m app.gradio_app