Skip to content

Entropy Stratified Regularized Adaptive Prediction Sets

Notifications You must be signed in to change notification settings

hex-research/entropy-conformal

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Setup

cd med-conformal

pip install -r requirements.txt

Run

python main.py

Configuration

Edit configs/config.yaml to customize:

experiment:
  name: "your_experiment_name"
  seed: 42

data:
  size: 224       
  batch_size: 128     
  
model:
  architecture: "resnet18"
  pretrained: true
  dropout: 0.3

training:
  epochs: 50
  learning_rate: 0.001
  early_stopping:
    enabled: true
    patience: 10

conformal:
  alpha: 0.1    
  methods:      
    - naive
    - lac
    - raps_size
    - raps_temp
    - raps_adaptive

gradcam:
  n_samples_correct: 20
  n_samples_incorrect: 20

About

Entropy Stratified Regularized Adaptive Prediction Sets

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%