Bert model is adapted from huggingface https://huggingface.co/transformers/model_doc/bert.html Bert fine-tuning codes are adapted from:https://mccormickml.com/2019/07/22/BERT-fine-tuning/
- Extracted sentences are available https://drive.google.com/drive/folders/1seAMJn3Fh8ZXhDpiW_ih7HDBlqHhIcnu?usp=sharing
- mitigation_model
- generate_neutral_latent_embedding_model
- bias_class_discriminator
./train_mitigation_model.sh "binary_classification_data.csv" "bias_only.train.en" "bias_only.dev.en" "neutral_only.train.en" "neutral_only.dev.en"
./mitigate_text.sh "bias_only.test.en" "mitigation_model.pth" "generate_neutral_latent_embedding_model.pth" "bias_class_discriminator.pth"
torch==1.12.1
transformers==4.21.2
lime==0.2.0.1
sentence_transformers==2.2.2
nltk==3.7
numpy