Index

A

AbstractProcessor (class in astrodata.data.processors.base) add() (astrodata.tracking.DataTracker.DataTracker method) add_remote() (astrodata.tracking.CodeTracker.CodeTracker method) add_to_index() (astrodata.tracking.CodeTracker.CodeTracker method) align_with_remote() (astrodata.tracking.CodeTracker.CodeTracker method) arbitrary_types_allowed (astrodata.data.schemas.ProcessedData.Config attribute) (astrodata.data.schemas.RawData.Config attribute) (astrodata.preml.schemas.Premldata.Config attribute) astrodata module astrodata.data module astrodata.data.loaders module astrodata.data.loaders.base module astrodata.data.loaders.csv_loader module astrodata.data.loaders.parquet_loader module astrodata.data.pipeline module astrodata.data.processors module astrodata.data.processors.base module astrodata.data.processors.common module astrodata.data.schemas module astrodata.data.utils 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.PytorchModel module astrodata.ml.models.SklearnModel module astrodata.ml.models.TensorflowModel 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.CodeTracker method) clone() (astrodata.ml.models.BaseMlModel.BaseMlModel method) CodeTracker (class in astrodata.tracking.CodeTracker)

config (astrodata.data.pipeline.DataPipeline attribute) convert_to_processed_data() (in module astrodata.data.utils) create_commit() (astrodata.tracking.CodeTracker.CodeTracker method) CsvLoader (class in astrodata.data.loaders.csv_loader)

D

data (astrodata.data.schemas.ProcessedData attribute), [1] (astrodata.data.schemas.RawData attribute), [1] DataPipeline (class in astrodata.data.pipeline) DataTracker (class in astrodata.tracking.DataTracker)

DropDuplicates (class in astrodata.data.processors.common) dump_parquet() (astrodata.data.schemas.ProcessedData method) (astrodata.preml.schemas.Premldata method) dump_supervised_ML_format() (astrodata.preml.schemas.Premldata method)

E

extract_format() (in module astrodata.data.utils)

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)

format (astrodata.data.schemas.RawData attribute), [1]

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) loader (astrodata.data.pipeline.DataPipeline attribute)

M

metadata (astrodata.data.schemas.ProcessedData attribute), [1] (astrodata.preml.schemas.Premldata attribute), [1] MissingImputator (class in astrodata.preml.processors.MissingImputator) MlflowBaseTracker (class in astrodata.tracking.MLFlowTracker) model_config (astrodata.data.schemas.ProcessedData attribute) (astrodata.data.schemas.RawData attribute) (astrodata.preml.schemas.Premldata attribute) ModelTracker (class in astrodata.tracking.ModelTracker) module astrodata astrodata.data astrodata.data.loaders astrodata.data.loaders.base astrodata.data.loaders.csv_loader astrodata.data.loaders.parquet_loader astrodata.data.pipeline astrodata.data.processors astrodata.data.processors.base astrodata.data.processors.common astrodata.data.schemas astrodata.data.utils 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.PytorchModel astrodata.ml.models.SklearnModel astrodata.ml.models.TensorflowModel 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

N

NormalizeAndSplit (class in astrodata.data.processors.common)

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.data.processors.base.AbstractProcessor method) (astrodata.data.processors.common.DropDuplicates method) (astrodata.data.processors.common.NormalizeAndSplit method) (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)

ProcessedData (class in astrodata.data.schemas) ProcessedData.Config (class in astrodata.data.schemas) processors (astrodata.data.pipeline.DataPipeline attribute) pull() (astrodata.tracking.CodeTracker.CodeTracker method) (astrodata.tracking.DataTracker.DataTracker method) push() (astrodata.tracking.CodeTracker.CodeTracker method) (astrodata.tracking.DataTracker.DataTracker method)

R

RawData (class in astrodata.data.schemas) RawData.Config (class in astrodata.data.schemas) read_config() (in module astrodata.utils.utils)

register_best_model() (astrodata.tracking.MLFlowTracker.MlflowBaseTracker method) remove_deleted_from_index() (astrodata.tracking.CodeTracker.CodeTracker method) run() (astrodata.data.pipeline.DataPipeline method) (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.MLFlowTracker) SklearnModel (class in astrodata.ml.models.SklearnModel) source (astrodata.data.schemas.RawData attribute), [1] Standardizer (class in astrodata.preml.processors.Standardizer)

T

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

Tracker (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.MLFlowTracker.MlflowBaseTracker method) (astrodata.tracking.MLFlowTracker.SklearnMLflowTracker method) (astrodata.tracking.ModelTracker.ModelTracker method)

X

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