Index A | B | C | D | E | F | G | I | K | L | M | N | O | P | Q | R | S | T | U | V | W A AbstractPreprocessor (class in pollux.data) AbstractSingleTransform (class in pollux.models) (class in pollux.models.transforms) AbstractTransform (class in pollux.models) (class in pollux.models.transforms) AdditiveOffsetTransform (class in pollux.models) (class in pollux.models.transforms) AffineTransform (class in pollux.models) (class in pollux.models.transforms) apply() (pollux.models.AbstractSingleTransform method) (pollux.models.AbstractTransform method) (pollux.models.AdditiveOffsetTransform method) (pollux.models.AffineTransform method) (pollux.models.EquinoxNNTransform method) (pollux.models.FunctionTransform method) (pollux.models.LinearTransform method) (pollux.models.NoOpTransform method) (pollux.models.OffsetTransform method) (pollux.models.PolyFeatureTransform method) (pollux.models.QuadraticTransform method) (pollux.models.transforms.AbstractSingleTransform method) (pollux.models.transforms.AbstractTransform method) (pollux.models.transforms.AdditiveOffsetTransform method) (pollux.models.transforms.AffineTransform method) (pollux.models.transforms.EquinoxNNTransform method) (pollux.models.transforms.FunctionTransform method) (pollux.models.transforms.LinearTransform method) (pollux.models.transforms.NoOpTransform method) (pollux.models.transforms.OffsetTransform method) (pollux.models.transforms.PolyFeatureTransform method) (pollux.models.transforms.QuadraticTransform method) (pollux.models.transforms.TransformSequence method) (pollux.models.TransformSequence method) B base_transform (pollux.models.AdditiveOffsetTransform attribute) (pollux.models.transforms.AdditiveOffsetTransform attribute) bias_prior (pollux.models.EquinoxNNTransform attribute) (pollux.models.transforms.EquinoxNNTransform attribute) C Cannon (class in pollux.models) coeffs (pollux.models.Cannon attribute), [1] D data (pollux.data.OutputData attribute) default_numpyro_model() (pollux.models.Lux method) (pollux.models.LuxModel method) degree (pollux.models.PolyFeatureTransform attribute) (pollux.models.transforms.PolyFeatureTransform attribute) dims (pollux.models.ShapeSpec attribute) (pollux.models.transforms.ShapeSpec attribute) E EquinoxNNTransform (class in pollux.models) (class in pollux.models.transforms) err (pollux.data.OutputData attribute) F fit() (pollux.models.Cannon method) from_data() (pollux.data.AbstractPreprocessor class method) (pollux.data.NullPreprocessor class method) (pollux.data.ShiftScalePreprocessor class method) from_data_percentiles() (pollux.data.ShiftScalePreprocessor class method) FunctionTransform (class in pollux.models) (class in pollux.models.transforms) G get() (pollux.data.PolluxData method) get_coeffs_as_transform_pars() (pollux.models.Cannon method) get_expanded_priors() (pollux.models.AbstractSingleTransform method) (pollux.models.AbstractTransform method) (pollux.models.AdditiveOffsetTransform method) (pollux.models.AffineTransform method) (pollux.models.EquinoxNNTransform method) (pollux.models.FunctionTransform method) (pollux.models.LinearTransform method) (pollux.models.NoOpTransform method) (pollux.models.OffsetTransform method) (pollux.models.PolyFeatureTransform method) (pollux.models.QuadraticTransform method) (pollux.models.transforms.AbstractSingleTransform method) (pollux.models.transforms.AbstractTransform method) (pollux.models.transforms.AdditiveOffsetTransform method) (pollux.models.transforms.AffineTransform method) (pollux.models.transforms.EquinoxNNTransform method) (pollux.models.transforms.FunctionTransform method) (pollux.models.transforms.LinearTransform method) (pollux.models.transforms.NoOpTransform method) (pollux.models.transforms.OffsetTransform method) (pollux.models.transforms.PolyFeatureTransform method) (pollux.models.transforms.QuadraticTransform method) (pollux.models.transforms.TransformSequence method) (pollux.models.TransformSequence method) get_features() (pollux.models.Cannon method) get_output_size() (pollux.models.PolyFeatureTransform method) (pollux.models.transforms.PolyFeatureTransform method) I include_bias (pollux.models.Cannon attribute) (pollux.models.PolyFeatureTransform attribute) (pollux.models.transforms.PolyFeatureTransform attribute) inverse_transform() (pollux.data.AbstractPreprocessor method) (pollux.data.NullPreprocessor method) (pollux.data.ShiftScalePreprocessor method) inverse_transform_err() (pollux.data.AbstractPreprocessor method) (pollux.data.NullPreprocessor method) (pollux.data.ShiftScalePreprocessor method) is_fitted (pollux.models.Cannon property) items() (pollux.data.PolluxData method) K keys() (pollux.data.PolluxData method) L label_size (pollux.models.Cannon attribute) latent_size (pollux.models.Lux attribute) (pollux.models.LuxModel attribute) LinearTransform (class in pollux.models) (class in pollux.models.transforms) loc (pollux.data.ShiftScalePreprocessor attribute) Lux (class in pollux.models) LuxModel (class in pollux.models) M module pollux.data pollux.models pollux.models.transforms N n_features (pollux.models.Cannon attribute) (pollux.models.Cannon property) names_flat (pollux.models.transforms.TransformSequence property) (pollux.models.TransformSequence property) names_nested (pollux.models.transforms.TransformSequence property) (pollux.models.TransformSequence property) nn_factory (pollux.models.EquinoxNNTransform attribute) (pollux.models.transforms.EquinoxNNTransform attribute) NoOpTransform (class in pollux.models) (class in pollux.models.transforms) NullPreprocessor (class in pollux.data) O offset_prior (pollux.models.AdditiveOffsetTransform attribute) (pollux.models.transforms.AdditiveOffsetTransform attribute) OffsetTransform (class in pollux.models) (class in pollux.models.transforms) optimize() (pollux.models.Lux method) (pollux.models.LuxModel method) optimize_iterative() (in module pollux.models) (pollux.models.Lux method) (pollux.models.LuxModel method) output_size (pollux.models.AbstractSingleTransform attribute) (pollux.models.AbstractTransform attribute) (pollux.models.AdditiveOffsetTransform property) (pollux.models.AffineTransform attribute) (pollux.models.Cannon attribute) (pollux.models.EquinoxNNTransform attribute) (pollux.models.FunctionTransform attribute) (pollux.models.LinearTransform attribute) (pollux.models.NoOpTransform attribute) (pollux.models.OffsetTransform attribute) (pollux.models.PolyFeatureTransform attribute) (pollux.models.QuadraticTransform attribute) (pollux.models.transforms.AbstractSingleTransform attribute) (pollux.models.transforms.AbstractTransform attribute) (pollux.models.transforms.AdditiveOffsetTransform property) (pollux.models.transforms.AffineTransform attribute) (pollux.models.transforms.EquinoxNNTransform attribute) (pollux.models.transforms.FunctionTransform attribute) (pollux.models.transforms.LinearTransform attribute) (pollux.models.transforms.NoOpTransform attribute) (pollux.models.transforms.OffsetTransform attribute) (pollux.models.transforms.PolyFeatureTransform attribute) (pollux.models.transforms.QuadraticTransform attribute) (pollux.models.transforms.TransformSequence attribute) (pollux.models.TransformSequence attribute) OutputData (class in pollux.data) outputs (pollux.models.Lux attribute) (pollux.models.LuxModel attribute) P pack_numpyro_pars() (pollux.models.Lux method) (pollux.models.LuxModel method) pack_pars() (pollux.models.AbstractSingleTransform method) (pollux.models.AdditiveOffsetTransform method) (pollux.models.AffineTransform method) (pollux.models.EquinoxNNTransform method) (pollux.models.FunctionTransform method) (pollux.models.LinearTransform method) (pollux.models.NoOpTransform method) (pollux.models.OffsetTransform method) (pollux.models.PolyFeatureTransform method) (pollux.models.QuadraticTransform method) (pollux.models.transforms.AbstractSingleTransform method) (pollux.models.transforms.AdditiveOffsetTransform method) (pollux.models.transforms.AffineTransform method) (pollux.models.transforms.EquinoxNNTransform method) (pollux.models.transforms.FunctionTransform method) (pollux.models.transforms.LinearTransform method) (pollux.models.transforms.NoOpTransform method) (pollux.models.transforms.OffsetTransform method) (pollux.models.transforms.PolyFeatureTransform method) (pollux.models.transforms.QuadraticTransform method) (pollux.models.transforms.TransformSequence method) (pollux.models.TransformSequence method) param_priors (pollux.models.AbstractSingleTransform attribute) (pollux.models.AffineTransform attribute) (pollux.models.EquinoxNNTransform property) (pollux.models.FunctionTransform attribute) (pollux.models.LinearTransform attribute) (pollux.models.NoOpTransform attribute) (pollux.models.OffsetTransform attribute) (pollux.models.PolyFeatureTransform property) (pollux.models.QuadraticTransform attribute) (pollux.models.transforms.AbstractSingleTransform attribute) (pollux.models.transforms.AffineTransform attribute) (pollux.models.transforms.EquinoxNNTransform property) (pollux.models.transforms.FunctionTransform attribute) (pollux.models.transforms.LinearTransform attribute) (pollux.models.transforms.NoOpTransform attribute) (pollux.models.transforms.OffsetTransform attribute) (pollux.models.transforms.PolyFeatureTransform property) (pollux.models.transforms.QuadraticTransform attribute) (pollux.models.transforms.TransformSequence property) (pollux.models.TransformSequence property) param_shapes (pollux.models.AbstractSingleTransform attribute) (pollux.models.AffineTransform attribute) (pollux.models.EquinoxNNTransform property) (pollux.models.FunctionTransform attribute) (pollux.models.LinearTransform attribute) (pollux.models.NoOpTransform attribute) (pollux.models.OffsetTransform attribute) (pollux.models.PolyFeatureTransform property) (pollux.models.QuadraticTransform attribute) (pollux.models.transforms.AbstractSingleTransform attribute) (pollux.models.transforms.AffineTransform attribute) (pollux.models.transforms.EquinoxNNTransform property) (pollux.models.transforms.FunctionTransform attribute) (pollux.models.transforms.LinearTransform attribute) (pollux.models.transforms.NoOpTransform attribute) (pollux.models.transforms.OffsetTransform attribute) (pollux.models.transforms.PolyFeatureTransform property) (pollux.models.transforms.QuadraticTransform attribute) (pollux.models.transforms.TransformSequence property) (pollux.models.TransformSequence property) ParamPriorsT (in module pollux.models.transforms) ParamShapesT (in module pollux.models.transforms) pollux.data module pollux.models module pollux.models.transforms module PolluxData (class in pollux.data) poly_degree (pollux.models.Cannon attribute) PolyFeatureTransform (class in pollux.models) (class in pollux.models.transforms) predict() (pollux.models.Cannon method) predict_outputs() (pollux.models.Lux method) (pollux.models.LuxModel method) preprocess() (pollux.data.OutputData method) (pollux.data.PolluxData method) preprocessor (pollux.data.OutputData attribute) priors (pollux.models.AbstractSingleTransform attribute) (pollux.models.AffineTransform attribute) (pollux.models.EquinoxNNTransform attribute) (pollux.models.FunctionTransform attribute) (pollux.models.LinearTransform attribute) (pollux.models.NoOpTransform attribute) (pollux.models.OffsetTransform attribute) (pollux.models.PolyFeatureTransform attribute) (pollux.models.QuadraticTransform attribute) (pollux.models.transforms.AbstractSingleTransform attribute) (pollux.models.transforms.AffineTransform attribute) (pollux.models.transforms.EquinoxNNTransform attribute) (pollux.models.transforms.FunctionTransform attribute) (pollux.models.transforms.LinearTransform attribute) (pollux.models.transforms.NoOpTransform attribute) (pollux.models.transforms.OffsetTransform attribute) (pollux.models.transforms.PolyFeatureTransform attribute) (pollux.models.transforms.QuadraticTransform attribute) (pollux.models.transforms.TransformSequence property) (pollux.models.TransformSequence property) processed (pollux.data.OutputData attribute) Q QuadraticTransform (class in pollux.models) (class in pollux.models.transforms) R register_output() (pollux.models.Lux method) (pollux.models.LuxModel method) resolve() (pollux.models.ShapeSpec method) (pollux.models.transforms.ShapeSpec method) S scale (pollux.data.ShiftScalePreprocessor attribute) scatter (pollux.models.Cannon attribute), [1] setup_numpyro() (pollux.models.Lux method) (pollux.models.LuxModel method) shapes (pollux.models.AbstractSingleTransform attribute) (pollux.models.AffineTransform attribute) (pollux.models.EquinoxNNTransform attribute) (pollux.models.FunctionTransform attribute) (pollux.models.LinearTransform attribute) (pollux.models.NoOpTransform attribute) (pollux.models.OffsetTransform attribute) (pollux.models.PolyFeatureTransform attribute) (pollux.models.QuadraticTransform attribute) (pollux.models.transforms.AbstractSingleTransform attribute) (pollux.models.transforms.AffineTransform attribute) (pollux.models.transforms.EquinoxNNTransform attribute) (pollux.models.transforms.FunctionTransform attribute) (pollux.models.transforms.LinearTransform attribute) (pollux.models.transforms.NoOpTransform attribute) (pollux.models.transforms.OffsetTransform attribute) (pollux.models.transforms.PolyFeatureTransform attribute) (pollux.models.transforms.QuadraticTransform attribute) (pollux.models.transforms.TransformSequence property) (pollux.models.TransformSequence property) ShapeSpec (class in pollux.models) (class in pollux.models.transforms) ShiftScalePreprocessor (class in pollux.data) T to_transform_sequence() (pollux.models.Cannon method) transform (pollux.models.AbstractSingleTransform attribute) (pollux.models.FunctionTransform attribute) (pollux.models.transforms.AbstractSingleTransform attribute) (pollux.models.transforms.FunctionTransform attribute) transform() (pollux.data.AbstractPreprocessor method) (pollux.data.NullPreprocessor method) (pollux.data.ShiftScalePreprocessor method) (pollux.models.AffineTransform method) (pollux.models.LinearTransform method) (pollux.models.NoOpTransform method) (pollux.models.OffsetTransform method) (pollux.models.QuadraticTransform method) (pollux.models.transforms.AffineTransform method) (pollux.models.transforms.LinearTransform method) (pollux.models.transforms.NoOpTransform method) (pollux.models.transforms.OffsetTransform method) (pollux.models.transforms.QuadraticTransform method) transform_err() (pollux.data.AbstractPreprocessor method) (pollux.data.NullPreprocessor method) (pollux.data.ShiftScalePreprocessor method) transforms (pollux.models.transforms.TransformSequence attribute) (pollux.models.TransformSequence attribute) TransformSequence (class in pollux.models) (class in pollux.models.transforms) tree_flatten() (pollux.data.PolluxData method) tree_unflatten() (pollux.data.PolluxData class method) U unpack_numpyro_pars() (pollux.models.Lux method) (pollux.models.LuxModel method) unpack_pars() (pollux.models.AbstractSingleTransform method) (pollux.models.AdditiveOffsetTransform method) (pollux.models.AffineTransform method) (pollux.models.EquinoxNNTransform method) (pollux.models.FunctionTransform method) (pollux.models.LinearTransform method) (pollux.models.NoOpTransform method) (pollux.models.OffsetTransform method) (pollux.models.PolyFeatureTransform method) (pollux.models.QuadraticTransform method) (pollux.models.transforms.AbstractSingleTransform method) (pollux.models.transforms.AdditiveOffsetTransform method) (pollux.models.transforms.AffineTransform method) (pollux.models.transforms.EquinoxNNTransform method) (pollux.models.transforms.FunctionTransform method) (pollux.models.transforms.LinearTransform method) (pollux.models.transforms.NoOpTransform method) (pollux.models.transforms.OffsetTransform method) (pollux.models.transforms.PolyFeatureTransform method) (pollux.models.transforms.QuadraticTransform method) (pollux.models.transforms.TransformSequence method) (pollux.models.TransformSequence method) unprocess() (pollux.data.OutputData method) (pollux.data.PolluxData method) V values() (pollux.data.PolluxData method) vmap (pollux.models.AbstractSingleTransform attribute) (pollux.models.AffineTransform attribute) (pollux.models.FunctionTransform attribute) (pollux.models.LinearTransform attribute) (pollux.models.NoOpTransform attribute) (pollux.models.OffsetTransform attribute) (pollux.models.QuadraticTransform attribute) (pollux.models.transforms.AbstractSingleTransform attribute) (pollux.models.transforms.AffineTransform attribute) (pollux.models.transforms.FunctionTransform attribute) (pollux.models.transforms.LinearTransform attribute) (pollux.models.transforms.NoOpTransform attribute) (pollux.models.transforms.OffsetTransform attribute) (pollux.models.transforms.QuadraticTransform attribute) W weight_prior (pollux.models.EquinoxNNTransform attribute) (pollux.models.transforms.EquinoxNNTransform attribute)
