Index _ | A | B | E | F | G | K | L | M | N | O | P | R | S | T | U | W _ __eq__() (metalog_jax.metalog.Metalog method) (metalog_jax.metalog.SPTMetalog method) __init__() (metalog_jax.metalog.GridResult method) (metalog_jax.metalog.Metalog method) (metalog_jax.metalog.SPTMetalog method) (metalog_jax.regression.base.RegressionModel method) (metalog_jax.regression.base.RegularizedParameters method) (metalog_jax.regression.lasso.LassoModel method) (metalog_jax.regression.lasso.LassoParameters method) (metalog_jax.regression.ols.OLSModel method) A a (metalog_jax.metalog.Metalog attribute), [1] (metalog_jax.metalog.SPTMetalog attribute), [1] B bias (metalog_jax.regression.lasso.LassoModel attribute) (metalog_jax.regression.ols.OLSModel attribute) E extract_best_from_grid() (in module metalog_jax.grid_search) extract_metalog() (in module metalog_jax.grid_search) F find_best_config() (in module metalog_jax.grid_search) fit() (in module metalog_jax.metalog) fit_grid() (in module metalog_jax.grid_search) fit_grid_datasets() (in module metalog_jax.grid_search) fit_grid_datasets_hyperparams() (in module metalog_jax.grid_search) fit_grid_datasets_num_terms() (in module metalog_jax.grid_search) fit_grid_full() (in module metalog_jax.grid_search) fit_grid_hyperparams() (in module metalog_jax.grid_search) fit_grid_num_terms() (in module metalog_jax.grid_search) fit_lasso() (in module metalog_jax.regression.lasso) fit_ordinary_least_squares() (in module metalog_jax.regression.ols) fit_spt_metalog() (in module metalog_jax.metalog) G GridResult (class in metalog_jax.metalog) K ks_dist (metalog_jax.metalog.GridResult attribute), [1] L lam (metalog_jax.regression.lasso.LassoParameters attribute), [1] LassoModel (class in metalog_jax.regression.lasso) LassoParameters (class in metalog_jax.regression.lasso) learning_rate (metalog_jax.regression.lasso.LassoParameters attribute), [1] M make_batch() (in module metalog_jax.grid_search) Metalog (class in metalog_jax.metalog) metalog (metalog_jax.metalog.GridResult attribute), [1] metalog_jax module metalog_jax.grid_search module metalog_jax.metalog module metalog_jax.regression.base module metalog_jax.regression.lasso module metalog_jax.regression.ols module metalog_params (metalog_jax.metalog.Metalog attribute), [1] (metalog_jax.metalog.SPTMetalog attribute), [1] method (metalog_jax.metalog.Metalog property) module metalog_jax metalog_jax.grid_search metalog_jax.metalog metalog_jax.regression.base metalog_jax.regression.lasso metalog_jax.regression.ols momentum (metalog_jax.regression.lasso.LassoParameters attribute), [1] N num_iters (metalog_jax.regression.lasso.LassoParameters attribute), [1] num_terms (metalog_jax.metalog.Metalog property) (metalog_jax.metalog.SPTMetalog property) O OLSModel (class in metalog_jax.regression.ols) P pad_metalog_coeffs() (in module metalog_jax.grid_search) predict_ordinary_least_squares() (in module metalog_jax.regression.ols) R RegressionModel (class in metalog_jax.regression.base) RegularizedParameters (class in metalog_jax.regression.base) replace() (metalog_jax.metalog.GridResult method) (metalog_jax.metalog.Metalog method) (metalog_jax.metalog.SPTMetalog method) (metalog_jax.regression.base.RegressionModel method) (metalog_jax.regression.base.RegularizedParameters method) (metalog_jax.regression.lasso.LassoModel method) (metalog_jax.regression.lasso.LassoParameters method) (metalog_jax.regression.ols.OLSModel method) S soft_thresholding() (in module metalog_jax.regression.lasso) SPTMetalog (class in metalog_jax.metalog) stack_leaves() (in module metalog_jax.grid_search) T tol (metalog_jax.regression.lasso.LassoParameters attribute), [1] U unvmap() (in module metalog_jax.grid_search) W weights (metalog_jax.regression.base.RegressionModel attribute), [1] (metalog_jax.regression.lasso.LassoModel attribute) (metalog_jax.regression.ols.OLSModel attribute)