Index

A

add() (astrodata.tracking.DataTracker method) (astrodata.tracking.DataTracker.DataTracker method) add_remote() (astrodata.tracking.CodeTracker method) (astrodata.tracking.CodeTracker.CodeTracker method) add_to_index() (astrodata.tracking.CodeTracker method) (astrodata.tracking.CodeTracker.CodeTracker method) align_with_remote() (astrodata.tracking.CodeTracker method) (astrodata.tracking.CodeTracker.CodeTracker method) arbitrary_types_allowed (astrodata.preml.schemas.Premldata.Config attribute) astrodata module astrodata.data.loaders.base module astrodata.data.loaders.csv_loader module astrodata.data.loaders.parquet_loader module astrodata.data.schemas module astrodata.ml module astrodata.ml.metrics module astrodata.ml.metrics.BaseMetric module astrodata.ml.metrics.SklearnMetric module astrodata.ml.model_selection module astrodata.ml.model_selection.BaseMlModelSelector module astrodata.ml.model_selection.GridSearchSelector module astrodata.ml.model_selection.HyperOptSelector module astrodata.ml.models module astrodata.ml.models.BaseMlModel module astrodata.ml.models.SklearnModel module

astrodata.ml.models.XGBoostModel module astrodata.preml module astrodata.preml.pipeline module astrodata.preml.processors module astrodata.preml.processors.base module astrodata.preml.processors.MissingImputator module astrodata.preml.processors.Ohe module astrodata.preml.processors.Standardizer module astrodata.preml.processors.TrainTestSplitter module astrodata.preml.schemas module astrodata.preml.utils module astrodata.tracking module astrodata.tracking.CodeTracker module astrodata.tracking.DataTracker module astrodata.tracking.MLFlowTracker module astrodata.tracking.ModelTracker module astrodata.tracking.Tracker module astrodata.utils module astrodata.utils.logger module astrodata.utils.utils module

B

BaseLoader (class in astrodata.data.loaders.base) BaseMetric (class in astrodata.ml.metrics.BaseMetric)

BaseMlModel (class in astrodata.ml.models.BaseMlModel) BaseMlModelSelector (class in astrodata.ml.model_selection.BaseMlModelSelector)

C

checkout() (astrodata.tracking.CodeTracker method) (astrodata.tracking.CodeTracker.CodeTracker method) clone() (astrodata.ml.models.BaseMlModel.BaseMlModel method) (astrodata.ml.models.SklearnModel.SklearnModel method) (astrodata.ml.models.XGBoostModel.XGBoostModel method)

CodeTracker (class in astrodata.tracking) (class in astrodata.tracking.CodeTracker) create_commit() (astrodata.tracking.CodeTracker method) (astrodata.tracking.CodeTracker.CodeTracker method) CsvLoader (class in astrodata.data.loaders.csv_loader)

D

DataTracker (class in astrodata.tracking) (class in astrodata.tracking.DataTracker)

dump_parquet() (astrodata.preml.schemas.Premldata method) dump_supervised_ML_format() (astrodata.preml.schemas.Premldata method)

F

fit() (astrodata.ml.model_selection.BaseMlModelSelector.BaseMlModelSelector method) (astrodata.ml.model_selection.GridSearchSelector.GridSearchCVSelector method) (astrodata.ml.model_selection.GridSearchSelector.GridSearchSelector method) (astrodata.ml.model_selection.HyperOptSelector.HyperOptSelector method) (astrodata.ml.models.BaseMlModel.BaseMlModel method) (astrodata.ml.models.SklearnModel.SklearnModel method) (astrodata.ml.models.XGBoostModel.XGBoostModel method)

G

get_best_metrics() (astrodata.ml.model_selection.BaseMlModelSelector.BaseMlModelSelector method) (astrodata.ml.model_selection.GridSearchSelector.GridSearchCVSelector method) (astrodata.ml.model_selection.GridSearchSelector.GridSearchSelector method) (astrodata.ml.model_selection.HyperOptSelector.HyperOptSelector method) get_best_model() (astrodata.ml.model_selection.BaseMlModelSelector.BaseMlModelSelector method) (astrodata.ml.model_selection.GridSearchSelector.GridSearchCVSelector method) (astrodata.ml.model_selection.GridSearchSelector.GridSearchSelector method) (astrodata.ml.model_selection.HyperOptSelector.HyperOptSelector method) get_best_params() (astrodata.ml.model_selection.BaseMlModelSelector.BaseMlModelSelector method) (astrodata.ml.model_selection.GridSearchSelector.GridSearchCVSelector method) (astrodata.ml.model_selection.GridSearchSelector.GridSearchSelector method) (astrodata.ml.model_selection.HyperOptSelector.HyperOptSelector method) get_loss_history() (astrodata.ml.models.SklearnModel.SklearnModel method) (astrodata.ml.models.XGBoostModel.XGBoostModel method) get_loss_history_metrics() (astrodata.ml.models.SklearnModel.SklearnModel method) (astrodata.ml.models.XGBoostModel.XGBoostModel method) get_metrics() (astrodata.ml.models.BaseMlModel.BaseMlModel method) (astrodata.ml.models.SklearnModel.SklearnModel method) (astrodata.ml.models.XGBoostModel.XGBoostModel method)

get_name() (astrodata.ml.metrics.BaseMetric.BaseMetric method) (astrodata.ml.metrics.SklearnMetric.SklearnMetric method) get_output_path() (in module astrodata.utils.utils) get_params() (astrodata.ml.model_selection.BaseMlModelSelector.BaseMlModelSelector method) (astrodata.ml.model_selection.GridSearchSelector.GridSearchCVSelector method) (astrodata.ml.model_selection.GridSearchSelector.GridSearchSelector method) (astrodata.ml.model_selection.HyperOptSelector.HyperOptSelector method) (astrodata.ml.models.BaseMlModel.BaseMlModel method) (astrodata.ml.models.SklearnModel.SklearnModel method) (astrodata.ml.models.XGBoostModel.XGBoostModel method) get_scorer_metric() (astrodata.ml.models.BaseMlModel.BaseMlModel method) (astrodata.ml.models.SklearnModel.SklearnModel method) (astrodata.ml.models.XGBoostModel.XGBoostModel method) git_operation() (in module astrodata.tracking.CodeTracker) greater_is_better (astrodata.ml.metrics.BaseMetric.BaseMetric property) (astrodata.ml.metrics.SklearnMetric.SklearnMetric property) GridSearchCVSelector (class in astrodata.ml.model_selection.GridSearchSelector) GridSearchSelector (class in astrodata.ml.model_selection.GridSearchSelector)

H

has_loss_history (astrodata.ml.models.SklearnModel.SklearnModel property) (astrodata.ml.models.XGBoostModel.XGBoostModel property)

HyperOptSelector (class in astrodata.ml.model_selection.HyperOptSelector)

I

instantiate_processors() (in module astrodata.preml.utils)

L

load() (astrodata.data.loaders.base.BaseLoader method) (astrodata.data.loaders.csv_loader.CsvLoader method) (astrodata.data.loaders.parquet_loader.ParquetLoader method) (astrodata.ml.models.BaseMlModel.BaseMlModel method) (astrodata.ml.models.SklearnModel.SklearnModel method) (astrodata.ml.models.XGBoostModel.XGBoostModel method)

load_artifact() (astrodata.preml.processors.base.PremlProcessor method)

M

metadata (astrodata.preml.schemas.Premldata attribute), [1] MissingImputator (class in astrodata.preml.processors.MissingImputator) MlflowBaseTracker (class in astrodata.tracking) (class in astrodata.tracking.MLFlowTracker) model_config (astrodata.preml.schemas.Premldata attribute) ModelTracker (class in astrodata.tracking) (class in astrodata.tracking.ModelTracker) module astrodata astrodata.data.loaders.base astrodata.data.loaders.csv_loader astrodata.data.loaders.parquet_loader astrodata.data.schemas astrodata.ml astrodata.ml.metrics astrodata.ml.metrics.BaseMetric astrodata.ml.metrics.SklearnMetric astrodata.ml.model_selection astrodata.ml.model_selection.BaseMlModelSelector astrodata.ml.model_selection.GridSearchSelector astrodata.ml.model_selection.HyperOptSelector astrodata.ml.models astrodata.ml.models.BaseMlModel astrodata.ml.models.SklearnModel astrodata.ml.models.XGBoostModel astrodata.preml astrodata.preml.pipeline astrodata.preml.processors astrodata.preml.processors.base astrodata.preml.processors.MissingImputator astrodata.preml.processors.Ohe astrodata.preml.processors.Standardizer astrodata.preml.processors.TrainTestSplitter astrodata.preml.schemas astrodata.preml.utils astrodata.tracking astrodata.tracking.CodeTracker astrodata.tracking.DataTracker astrodata.tracking.MLFlowTracker astrodata.tracking.ModelTracker astrodata.tracking.Tracker astrodata.utils astrodata.utils.logger astrodata.utils.utils

O

OHE (class in astrodata.preml.processors.Ohe)

P

ParquetLoader (class in astrodata.data.loaders.parquet_loader) predict() (astrodata.ml.models.BaseMlModel.BaseMlModel method) (astrodata.ml.models.SklearnModel.SklearnModel method) (astrodata.ml.models.XGBoostModel.XGBoostModel method) predict_proba() (astrodata.ml.models.SklearnModel.SklearnModel method) Premldata (class in astrodata.preml.schemas) Premldata.Config (class in astrodata.preml.schemas) PremlPipeline (class in astrodata.preml.pipeline) PremlProcessor (class in astrodata.preml.processors.base) process() (astrodata.preml.processors.base.PremlProcessor method) (astrodata.preml.processors.MissingImputator.MissingImputator method) (astrodata.preml.processors.Ohe.OHE method) (astrodata.preml.processors.Standardizer.Standardizer method) (astrodata.preml.processors.TrainTestSplitter.TrainTestSplitter method)

pull() (astrodata.tracking.CodeTracker method) (astrodata.tracking.CodeTracker.CodeTracker method) (astrodata.tracking.DataTracker method) (astrodata.tracking.DataTracker.DataTracker method) push() (astrodata.tracking.CodeTracker method) (astrodata.tracking.CodeTracker.CodeTracker method) (astrodata.tracking.DataTracker method) (astrodata.tracking.DataTracker.DataTracker method) PytorchMLflowTracker (class in astrodata.tracking) (class in astrodata.tracking.MLFlowTracker)

R

read_config() (in module astrodata.utils.utils) register_best_model() (astrodata.tracking.MlflowBaseTracker method) (astrodata.tracking.MLFlowTracker.MlflowBaseTracker method)

remove_deleted_from_index() (astrodata.tracking.CodeTracker method) (astrodata.tracking.CodeTracker.CodeTracker method) run() (astrodata.preml.pipeline.PremlPipeline method)

S

save() (astrodata.ml.models.BaseMlModel.BaseMlModel method) (astrodata.ml.models.SklearnModel.SklearnModel method) (astrodata.ml.models.XGBoostModel.XGBoostModel method) save_artifact() (astrodata.preml.processors.base.PremlProcessor method) score() (astrodata.ml.models.BaseMlModel.BaseMlModel method) (astrodata.ml.models.SklearnModel.SklearnModel method) (astrodata.ml.models.XGBoostModel.XGBoostModel method) set_params() (astrodata.ml.models.BaseMlModel.BaseMlModel method) (astrodata.ml.models.SklearnModel.SklearnModel method) (astrodata.ml.models.XGBoostModel.XGBoostModel method)

setup_logger() (in module astrodata.utils.logger) SklearnMetric (class in astrodata.ml.metrics.SklearnMetric) SklearnMLflowTracker (class in astrodata.tracking) (class in astrodata.tracking.MLFlowTracker) SklearnModel (class in astrodata.ml.models.SklearnModel) Standardizer (class in astrodata.preml.processors.Standardizer)

T

TensorflowMLflowTracker (class in astrodata.tracking) (class in astrodata.tracking.MLFlowTracker) test_features (astrodata.preml.schemas.Premldata attribute), [1] test_targets (astrodata.preml.schemas.Premldata attribute), [1] track() (astrodata.tracking.Tracker method) (astrodata.tracking.Tracker.Tracker method)

Tracker (class in astrodata.tracking) (class in astrodata.tracking.Tracker) train_features (astrodata.preml.schemas.Premldata attribute), [1] train_targets (astrodata.preml.schemas.Premldata attribute), [1] TrainTestSplitter (class in astrodata.preml.processors.TrainTestSplitter)

V

val_features (astrodata.preml.schemas.Premldata attribute), [1]

val_targets (astrodata.preml.schemas.Premldata attribute), [1]

W

wrap_fit() (astrodata.tracking.MlflowBaseTracker method) (astrodata.tracking.MLFlowTracker.MlflowBaseTracker method) (astrodata.tracking.MLFlowTracker.SklearnMLflowTracker method) (astrodata.tracking.ModelTracker method) (astrodata.tracking.ModelTracker.ModelTracker method) (astrodata.tracking.SklearnMLflowTracker method)

X

XGBoostModel (class in astrodata.ml.models.XGBoostModel)