Skip to content

Latest commit

 

History

History
41 lines (25 loc) · 1.17 KB

README.md

File metadata and controls

41 lines (25 loc) · 1.17 KB

Adaptive Dimension Reduction and Variational Inference for Transductive Few-Shot Classification

This repository is the official implementation of [Adaptive Dimension Reduction and Variational Inference for Transductive Few-Shot Classification].

Requirements

To install requirements:

pip install -r requirements.txt

Evaluation

To evaluate my model on CUB for the unbalanced setting, run:

python bavardage.py --shot 1/5 --T_km 10 --T_vb 4 --clp 5

For the balanced setting, run:

python bavardage.py --shot 1/5 --T_km 10 --T_vb 4 --clp 5 --balanced

Pre-trained Models

We provide a WRN model pretrained on CUB in ./checkpoints

Results

Our model achieves the following performance on :

CUB

Model name 1 shot (unbalanced) 5 shot (unbalanced) 1 shot (balanced) 5 shot (balanced)
BAVARDAGE 82.00% 90.67% 85.64% 91.42%