codyreading / syde522-data-challenge

Primary site classification in histopathology images

syde522-data-challenge

Primary site classification in histopathology images. Uses ResNet50 and train's from scratch with Keras.

Environment setup

Create an Anaconda environment for this project

conda env create -f environment.yml

Training

Run the following script to train the model

./src/train.py

Testing

Run the following script to test the model

./src/test.py