Index

A

AbstractProcessor (class in astrodata.data.processors.base) arbitrary_types_allowed (astrodata.data.schemas.ProcessedData.Config attribute) (astrodata.data.schemas.RawData.Config attribute) (astrodata.preml.schemas.Premldata.Config attribute) astrodata module, [1] astrodata.data module astrodata.data.loaders module astrodata.data.loaders.base 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.schemas module astrodata.preml.utils module astrodata.tracking 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

clone() (astrodata.ml.models.BaseMlModel.BaseMlModel method)

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

D

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

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_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) 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.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) log_metrics_and_loss() (in module astrodata.tracking.MLFlowTracker) log_metrics_manual() (in module astrodata.tracking.MLFlowTracker)

M

metadata (astrodata.data.schemas.ProcessedData attribute), [1] (astrodata.preml.schemas.Premldata attribute), [1] 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, [1] astrodata.data astrodata.data.loaders astrodata.data.loaders.base 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.schemas astrodata.preml.utils astrodata.tracking 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)

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) ProcessedData (class in astrodata.data.schemas) ProcessedData.Config (class in astrodata.data.schemas) processors (astrodata.data.pipeline.DataPipeline attribute)

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) 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]

T

test_features (astrodata.preml.schemas.Premldata attribute) test_targets (astrodata.preml.schemas.Premldata attribute) track() (astrodata.tracking.Tracker.Tracker method)

Tracker (class in astrodata.tracking.Tracker) train_features (astrodata.preml.schemas.Premldata attribute) train_targets (astrodata.preml.schemas.Premldata attribute)

V

val_features (astrodata.preml.schemas.Premldata attribute)

val_targets (astrodata.preml.schemas.Premldata attribute)

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)