Colin97/OpenShape_code

Few-shot linear probing implementation ?

alanzhangcs opened this issue · 1 comments

Hi, thanks for your great work! I'm wondering that could you release your implementation details of your Few-shot linear probing experiment ?

Our few-shot implementation follows closely to the one in 2D CLIP. Given that features of all 3D shapes in the test dataset are extracted, we randomly choose N samples from each category in the test data and use these features to train a logistic regression model. We simply use scikit-learn's Logistic Regression with LBFGS solver, as in 2D CLIP. Next, we evaluate the trained logistic model on the rest of the test dataset. This process is repeated 10 times and we take the average performance.