Performance on the test set of the linear regression model for each cosmological parameter, and for the three datasets, AVG100, AVG300, and AVG500, with θ_s = 2′. The dots represent individual predictions. The shaded regions delimit the 1 σ interval obtained dividing the test sample into 10 bins of the true label values and calculating the mean and standard deviation of the predictions. We obtain the best predictions with the AVG500 dataset and for the Ω_M, Ω_Λ, w₀, and σ₈ parameters.