Hypers at ComMA@ICON: Modelling Aggressiveness, Gender Bias and Communal Bias Identification
Paper
•
2112.15417
•
Published
This is a XLM-Align-Base model trained on CoMMA dataset of 12k samples
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)
@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}
}