Skip to content

shengtaihuang/fml_project

Repository files navigation

Detect Group Bias in COVID-19 Anti-vax Detection Models (2022 Spring Fairness in Machine Learning)

  • Team members: Sheng-Tai Huang
  • Project paper: link

{Below is just some basics information you should include in your project README}

Description

{A summary of what the project is and does, the technology it employs, and the purpose behind the project.} {Provide a summary for the major directories and files and what they do. You can also provide a table of content here and link to seperate readme files.}

  • Pre-trained COVID BERT: link
  • Labeled ANTivax Data: link

Folders:

  • data/: raw Facebook data of each group
  • labeled/: labeled ANTiVax data and my labeled Facebook data

Files

Python

  • bertopic.ipynb: Extract embeddings from the fine-tuned COVID-Twitter-BERT and use BERTopic for topic modeling
  • covid-bert_finetune.ipynb: Fine-tune the COVID-Twitter-BERT by using ANTiVax data and assess on my collected Facebook data

R

  • clean_not_group.R: filter out groups that is not aligned to the target keywords, for example, filter out groups with "conservative" but are actually opposing conservative
  • generate_labeled_data.R: Data pre-processing for hydrating data
  • plot_bertopic.R: Scatter plots of BERTopic

Results

bertopic_result.csv: Embeddings and topic modeling results from the fine-tuned COVID-Twitter-BERT

Prerequisites

{If your project requires any applications or special configurations, you will place them in this section. List any requirements, packages, configurations, etc. Add in any special requirements for file system or database.}

Packages:

Python R
hdbscan, json, nltk, numpy, pandas, re, sklearn, tensorflow, transformers, umap data.table, dplyr, ggplot2, jsonlite

Inspiration

Authors

Sheng-Tai Huang: shengtai.huang@pitt.edu

License

{Provide the license information, e.g.,} MIT

About

Course project of "Fate in Machine Learning"

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published