search

comsaint / dlh_project

For CS598 Deep Learning for Healthcare project
0 stars 0 forks source link

Working sample code for NIH #4

Closed comsaint closed 3 years ago

comsaint commented 3 years ago

This is a working template of a notebook which takes in the NIH dataset and train a CNN which classify 1 disease.

How to use:

  1. For the first run, uncomment the 2 !gsutil ... lines which downloads the image and labels.
  2. Run all cells. Some cells are for visualization.

TODO:

  1. Prettify the code for train and validation.
  2. Complete the evaluation using test set. Currently it is commentted out,