about trained model in dir 'ICI-FSL/ckpt'

Yikai-Wang / ICI-FSL

This repository contains the code for Instance Credibility Inference (ICI).

82 stars 22 forks source link

Closed xiaofeng-github closed 4 years ago

xiaofeng-github commented 4 years ago

hello, the saved models in dir 'ICI-FSL/ckpt' are best_model for correspondent dataset?

Yikai-Wang commented 4 years ago

The saved models are selected by https://github.com/Yikai-Wang/ICI-FSL/blob/f2537b7742159adfa68cc8646b9a20c570320443/main.py#L75-L80 with the best accuracy on the test set of base classes (split by EmbeddingDataset).