phobihsd-proposed / README.md
joshswift's picture
Upload PhoBiHSD pretrained checkpoint
ce2b24a verified
metadata
library_name: transformers
license: apache-2.0
tags:
  - vietnamese
  - hate-speech-detection
  - phobert
  - bilstm
  - text-classification
pipeline_tag: text-classification

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: Clean
  • 1: Offensive
  • 2: 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