nogibjj/Reddit_Classifier_Final_Project

Jupyter Notebook

0

13 commits

updated Dec 13, 2022

See the code

README

CI Codespaces Prebuilds

Get Reddit data

  • Data pulled in notebook reddit_data/reddit_new.ipynb

Verify GPU works

  • Run pytorch training test: python utils/quickstart_pytorch.py
  • Run pytorch CUDA test: python utils/verify_cuda_pytorch.py
  • Run tensorflow training test: python utils/quickstart_tf2.py
  • Run nvidia monitoring test: nvidia-smi -l 1

Finetune text classifier model and upload to Hugging Face

  • In terminal, run huggingface-cli login
  • Run python fine_tune_berft.py to finetune the model on Reddit data
  • Run rename_labels.py to change the output labels of the classifier
  • Check out the fine-tuned model here Spaces APP

nogibjj/Reddit_Classifier_Final_Project

Jupyter Notebook

0

13 commits

updated Dec 13, 2022

See the code

README

CI Codespaces Prebuilds

Get Reddit data

  • Data pulled in notebook reddit_data/reddit_new.ipynb

Verify GPU works

  • Run pytorch training test: python utils/quickstart_pytorch.py
  • Run pytorch CUDA test: python utils/verify_cuda_pytorch.py
  • Run tensorflow training test: python utils/quickstart_tf2.py
  • Run nvidia monitoring test: nvidia-smi -l 1

Finetune text classifier model and upload to Hugging Face

  • In terminal, run huggingface-cli login
  • Run python fine_tune_berft.py to finetune the model on Reddit data
  • Run rename_labels.py to change the output labels of the classifier
  • Check out the fine-tuned model here Spaces APP