Skip to content

davisgomes/CS-168-Final-Project

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

14 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Authors

Davis Gomes, Brian Lee, Yash Choudhary

CS-168-Final-Project

I have created this repository to host the code for the CS 168 Final Project. As we go along, this file will be updated.

General Info

I have created a deep learning model based off on the Pytorch Model. This has most of the info necessary to understand whats going on behind the scenes in this project. The implementation contains a few models (resnet, alexnet, etc.) to use in training and validation sets. The best validation error along with the weights is returned.

I have set up and argument parser with the model name, number of epochs, and batch size as arguments (Only model name is required). You can use the help option (-h) for more info. I was able to get 86% validation accuracy for resnet and 81% for alexnet. We can work on other methods from the paper if necessary or just use an array of models.

New arguments have been added, namely for pretraining and the split percentage. Setting the -p/--pretrain flag will set the model to be pretrained and -s/--split S_PER requires a float value for the precentage of data used for training.

Let me know if you have any questions about thew code or implementation and I'll try my best to answer!

GIT HELP:

initialize the directory as a git repo
  • git init
pull the repository
  • git pull # if it prompts a password try: CS168COVID
to upload files
  • git add . # or git add *
  • git commit -m "insert message"
if it troubles you make sure you are verified
  • git config --global user.email "[email protected]"
  • git config --global user.name "username"
and make sure the remote add is set up
  • git remote add CS-168-Final-Project "[email protected]:davisgomes/CS-168-Final-Project.git"
it will prompt you, use the password above: CS168COVID
push it to github
  • git push CS-168-Final-Project master

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages