Skip to content

Generator-Classifier Model for Interpretable Classification

Notifications You must be signed in to change notification settings

abhigoyal1997/gen-class

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

14 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Generator-Classifier Networks

Training Specifications

  1. BATCH_SIZE
  2. NUMBER_OF_EPOCHS
  3. TRAIN_RATIO
  4. RATIO_OF_MASKS
  5. NUMBER_OF_WORKERS (for data loader)

GenClassifier Configuration

  1. MODEL_CODE (genclassifier)
  2. NUMBER_OF_MASK_SAMPLES
  3. PATH_TO_GENERATOR
  4. PATH_TO_CLASSIFIER

Generator Configuration

  1. MODEL_CODE (generator)
  2. IMAGE_CHANNELS IMAGE_HEIGHT IMAGE_WIDTH
  3. ARCHITECTURE (one line for each module)

Classifier Configuration

  1. MODEL_CODE (classifier)
  2. IMAGE_CHANNELS IMAGE_HEIGHT IMAGE_WIDTH
  3. USE_MASKS CROP_SIZE
  4. ARCHITECTURE (one line for each module)

About

Generator-Classifier Model for Interpretable Classification

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages