/eigenlearning

codebase for Neural Tangent Kernel Eigenvalues Accurately Predict Generalization

Primary LanguageJupyter Notebook

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!