Skip to content

Add KNN classification to PILatentEvalVideo#484

Draft
ElmoPA wants to merge 1 commit into
mainfrom
elmo/eval-latent-knn
Draft

Add KNN classification to PILatentEvalVideo#484
ElmoPA wants to merge 1 commit into
mainfrom
elmo/eval-latent-knn

Conversation

@ElmoPA

@ElmoPA ElmoPA commented Jun 3, 2026

Copy link
Copy Markdown
Contributor

5-fold CV KNN classifier per layer slice predicting embodiment from key vectors. Uses cuML GPU when available, sklearn CPU fallback. New evaluator knobs compute_knn and knn_k. Logs accuracy at INFO so base vs finetuned diff surfaces in the run log directly.

5-fold CV KNN classifier per layer slice predicting embodiment from key vectors. Uses cuML GPU when available, sklearn CPU fallback. New evaluator knobs compute_knn and knn_k. Logs accuracy at INFO so base vs finetuned diff surfaces in the run log directly.

ElmoPA commented Jun 3, 2026

Copy link
Copy Markdown
Contributor Author

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant