Saved in:
Bibliographic Details
Main Author: Dubbs, Alexander
Format: Preprint
Published: 2025
Subjects:
Online Access:https://arxiv.org/abs/2504.19231
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909772213649408
author Dubbs, Alexander
author_facet Dubbs, Alexander
contents We derive the ideal train/test split for the ridge regression to high accuracy in the limit that the number of training rows m becomes large. The split must depend on the ridge tuning parameter, alpha, but we find that the dependence is weak and can asymptotically be ignored; all parameters vanish except for m and the number of features, n, which is held constant. This is the first time that such a split is calculated mathematically for a machine learning model in the large data limit. The goal of the calculations is to maximize "integrity," so that the measured error in the trained model is as close as possible to what it theoretically should be. This paper's result for the ridge regression split matches prior art for the plain vanilla linear regression split to the first two terms asymptotically.
format Preprint
id arxiv_https___arxiv_org_abs_2504_19231
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Test Set Sizing for the Ridge Regression
Dubbs, Alexander
Machine Learning
Probability
We derive the ideal train/test split for the ridge regression to high accuracy in the limit that the number of training rows m becomes large. The split must depend on the ridge tuning parameter, alpha, but we find that the dependence is weak and can asymptotically be ignored; all parameters vanish except for m and the number of features, n, which is held constant. This is the first time that such a split is calculated mathematically for a machine learning model in the large data limit. The goal of the calculations is to maximize "integrity," so that the measured error in the trained model is as close as possible to what it theoretically should be. This paper's result for the ridge regression split matches prior art for the plain vanilla linear regression split to the first two terms asymptotically.
title Test Set Sizing for the Ridge Regression
topic Machine Learning
Probability
url https://arxiv.org/abs/2504.19231