ONNX Model Zoo

seanbenhur/MuLTiGENBiaS

By seanbenhur · ↓ 5 · ♥ 0 · Updated 2023-09-11

What's in this repo

This page lists the files in seanbenhur/MuLTiGENBiaS, including pytorch_model.onnx.

  • 1 ONNX models (.onnx files)
  • 10 files, including 4 .json, 1 .md, 1 .safetensors, 1 .bin, 1 .onnx, 1 .h5
  • Last updated 2023-09-11

ONNX files

File Size Download
pytorch_model.onnx — Download

How to use

These ONNX files can be loaded with ONNX Runtime. To inspect a model graph, upload the file to the Netron viewer. Download links point to the original files on Hugging Face.

README

Automatic Identification of Gender Bias in Hindi,Bengali,Meitei Codemixed Texts

This is a XLM-Align-Base model trained on CoMMA dataset of 12k samples

Example Usage

import torch
import numpy as np
from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
from transformers import set_seed

set_seed(425)

text = "some gender biased text"
pipe = pipeline("text-classification", model="seanbenhur/MuLTiGENBiaS")

def predict_pipe(text):
    prediction = pipe(text, return_all_scores=True)[0]
    return prediction

if __name__ == "__main__":
  target = predict_pipe(text)
  print(target)
  

Some concerns

  • Note: The model is trained on relatively lower samples (i.e 12k) but with mix of four languages Hindi, Bengali, Meitei, and English. It contains both native on codemixed scripts, So the model might perform poorly on many text samples and might not generalize well.

Bibtex

 @article{Benhur2021HypersAC,
  title={Hypers at ComMA@ICON: Modelling Aggressiveness, Gender Bias and Communal Bias Identification},
  author={Sean Benhur and Roshan Nayak and Kanchana Sivanraju and Adeep Hande and Subalalitha Chinnaudayar Navaneethakrishnan and Ruba Priyadharshini and Bharathi Raja Chakravarthi6},
  journal={ArXiv},
  year={2021},
  volume={abs/2112.15417}
}

Model info

text-classification transformers
transformers pytorch tf onnx safetensors xlm-roberta text-classification Text Classification hn bn mn dataset:ComMA
↓ 5 ♥ 0

Created: 2022-03-02

Updated: 2023-09-11

View on Hugging Face

Files (10)

.gitattributes
README.md
config.json
model.safetensors
pytorch_model.bin
pytorch_model.onnx ONNX
special_tokens_map.json
tf_model.h5
tokenizer.json
tokenizer_config.json

Related models