WynMew / eigenlearning

codebase for Neural Tangent Kernel Eigenvalues Accurately Predict Generalization

Home Page:https://arxiv.org/abs/2110.03922

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Eigenlearning

This repo contains code for replicating the experiments of the paper NTK Eigenvalues Accurately Predict Generalization. The .py files provide a general codebase for generating synthetic datasets and testing the performance of kernel regression and finite nets learning functions over on the domains of interest. The .ipynb files contain code for generating the specific figures of the paper.

If you discover any bugs or want to request any changes, let us know!

About

codebase for Neural Tangent Kernel Eigenvalues Accurately Predict Generalization

https://arxiv.org/abs/2110.03922


Languages

Language:Jupyter Notebook 97.3%Language:Python 2.7%