Update README

README.md

@@ -19,77 +19,106 @@ This milestone includes finetuning a language model in HuggingFace for sentiment
 
 Link to app: https://huggingface.co/spaces/andyqin18/sentiment-analysis-app
 
-## 1. Space setup
-## 1. Space setup
+Here's the setup block:
+```
+import pandas as pd
+import numpy as np
+import torch
+from sklearn.model_selection import train_test_split
+from torch.utils.data import Dataset
+from transformers import AutoTokenizer, AutoModelForSequenceClassification, TrainingArguments, Trainer
+device = torch.device('cuda') if torch.cuda.is_available() else torch.device('cpu')
+```
+
+## 1. Prepare Data
+First we extract comment strings and labels from `train.csv` and split them into training data and validation data with a percentage of 80% vs 20%. We also create 2 dictionaries that map labels to integers and back.
+```
+df = pd.read_csv("milestone3/comp/train.csv")
+
+train_texts = df["comment_text"].values
+labels = df.columns[2:]
+id2label = {idx:label for idx, label in enumerate(labels)}
+label2id = {label:idx for idx, label in enumerate(labels)}
+train_labels = df[labels].values
+
 
-After creating a HuggingFace account, we can create our app as a space and choose Streamlit as the space SDK.
+# Randomly select 20000 samples within the data
+np.random.seed(18)
+small_train_texts = np.random.choice(train_texts, size=20000, replace=False)
 
-![](milestone2/new_HF_space.png)
+np.random.seed(18)
+small_train_labels_idx = np.random.choice(train_labels.shape[0], size=20000, replace=False)
+small_train_labels = train_labels[small_train_labels_idx, :]
 
-Then we can go back to our Github Repo and create the following files.
-In order for the space to run properly, there must be at least three files in the root directory: 
-[README.md](README.md), [app.py](app.py), and [requirements.txt](requirements.txt)
 
-Make sure the following metadata is at the top of **README.md** for HuggingFace to identify.
+# Separate training data and validation data
+train_texts, val_texts, train_labels, val_labels = train_test_split(small_train_texts, small_train_labels, test_size=.2)
 ```
----
-title: Sentiment Analysis App
-emoji: 🚀
-colorFrom: green
-colorTo: purple
-sdk: streamlit
-sdk_version: 1.17.0
-app_file: app.py
-pinned: false
----
+
+## 2. Data Preprocessing
+As models like BERT don't expect text as direct input, but rather `input_ids`, etc., we tokenize the text using the tokenizer. The `AutoTokenizer` will automatically load the appropriate tokenizer based on the checkpoint on the hub. We can now merge the labels and texts to datasets as a class we defined.
 ```
+tokenizer = AutoTokenizer.from_pretrained("bert-base-uncased")
 
-The **app.py** file is the main code of the app and **requirements.txt** should include all the libraries the code uses. HuggingFace will install the libraries listed before running the virtual environment
+class TextDataset(Dataset):
+  def __init__(self,texts,labels):
+    self.texts = texts
+    self.labels = labels
 
+  def __getitem__(self,idx):
+    encodings = tokenizer(self.texts[idx], truncation=True, padding="max_length")
+    item = {key: torch.tensor(val) for key, val in encodings.items()}
+    item['labels'] = torch.tensor(self.labels[idx],dtype=torch.float32)
+    del encodings
+    return item
 
-## 2. Connect and sync to HuggingFace
+  def __len__(self):
+    return len(self.labels)
 
-Then we go to settings of the Github Repo and create a secret token to access the new HuggingFace space. 
 
-![](milestone2/HF_token.png)
-![](milestone2/github_token.png)
+train_dataset = TextDataset(train_texts, train_labels)
+val_dataset = TextDataset(val_texts, val_labels)
+```
 
-Next, we need to setup a workflow in Github Actions. Click "set up a workflow yourself" and replace all the code in `main.yaml` with the following: (Replace `HF_USERNAME` and `SPACE_NAME` with our own)
+## 3. Train the model using Trainer
+We define a model that includes a pre-trained base and also set the problem to `multi_label_classification`. Then we train the model using `Trainer`, which requires  `TrainingArguments` beforehand that specify training hyperparameters, where we can set learning rate, batch sizes and `push_to_hub=True`.
+
+After verifying Token with HuggingFace, the model is now pushed to the hub.
 
 ```
-name: Sync to Hugging Face hub
-on:
-  push:
-    branches: [main]
-
-  # to run this workflow manually from the Actions tab
-  workflow_dispatch:
-
-jobs:
-  sync-to-hub:
-    runs-on: ubuntu-latest
-    steps:
-      - uses: actions/checkout@v3
-        with:
-          fetch-depth: 0
-          lfs: true
-      - name: Push to hub
-        env:
-          HF_TOKEN: ${{ secrets.HF_TOKEN }}
-        run: git push --force https://HF_USERNAME:[email protected]/spaces/HF_USERNAME/SPACE_NAME main
+model = AutoModelForSequenceClassification.from_pretrained("bert-base-uncased", 
+                                                           problem_type="multi_label_classification", 
+                                                           num_labels=len(labels),
+                                                           id2label=id2label,
+                                                           label2id=label2id)
+model.to(device)
+
+training_args = TrainingArguments(
+    output_dir="finetuned-bert-uncased",
+    evaluation_strategy = "epoch",
+    save_strategy = "epoch",
+    learning_rate=2e-5,
+    per_device_train_batch_size=16,
+    per_device_eval_batch_size=16,
+    num_train_epochs=5,
+    load_best_model_at_end=True,
+    push_to_hub=True
+)
+
+trainer = Trainer(
+    model=model,
+    args=training_args,
+    train_dataset=train_dataset,
+    eval_dataset=val_dataset,
+    tokenizer=tokenizer
+)
+
+trainer.train()
+trainer.push_to_hub()
 ```
-The Repo is now connected and synced with HuggingFace space!
 
-## 3. Create the app
+## 4. Update the app
 
 Modify [app.py](app.py) so that it takes in one text and generate an analysis using one of the provided models. Details are explained in comment lines. The app should look like this:
 
-![](milestone2/app_UI.png)
-
-
-## Reference:
-For connecting Github with HuggingFace, check this [video](https://www.youtube.com/watch?v=8hOzsFETm4I).
-
-For creating the app, check this [video](https://www.youtube.com/watch?v=GSt00_-0ncQ)
-
-The HuggingFace documentation is [here](https://huggingface.co/docs), and Streamlit APIs [here](https://docs.streamlit.io/library/api-reference).
+![](milestone3/appUI.png)

app.py

@@ -34,7 +34,7 @@ model_descrip = {
         Labels: POS; NEU; NEG"
 }
 
-user_input = st.text_input("Enter your text:", value="NYU is the better than Columbia.")
+user_input = st.text_input("Enter your text:", value="I hate NLP. Always lacking GPU.")
 user_model = st.selectbox("Please select a model:", model_descrip)