astrodata.ml package

Subpackages

• astrodata.ml.metrics package
  • Submodules
  • astrodata.ml.metrics.BaseMetric module
    • BaseMetric
      • BaseMetric.get_name()
      • BaseMetric.greater_is_better
  • astrodata.ml.metrics.SklearnMetric module
    • SklearnMetric
      • SklearnMetric.get_name()
      • SklearnMetric.greater_is_better
  • Module contents
• astrodata.ml.model_selection package
  • Submodules
  • astrodata.ml.model_selection.BaseMlModelSelector module
    • BaseMlModelSelector
      • BaseMlModelSelector.fit()
      • BaseMlModelSelector.get_best_metrics()
      • BaseMlModelSelector.get_best_model()
      • BaseMlModelSelector.get_best_params()
      • BaseMlModelSelector.get_params()
  • astrodata.ml.model_selection.GridSearchSelector module
    • GridSearchCVSelector
      • GridSearchCVSelector.fit()
      • GridSearchCVSelector.get_best_metrics()
      • GridSearchCVSelector.get_best_model()
      • GridSearchCVSelector.get_best_params()
      • GridSearchCVSelector.get_params()
    • GridSearchSelector
      • GridSearchSelector.fit()
      • GridSearchSelector.get_best_metrics()
      • GridSearchSelector.get_best_model()
      • GridSearchSelector.get_best_params()
      • GridSearchSelector.get_params()
  • astrodata.ml.model_selection.HyperOptSelector module
    • HyperOptSelector
      • HyperOptSelector.fit()
      • HyperOptSelector.get_best_metrics()
      • HyperOptSelector.get_best_model()
      • HyperOptSelector.get_best_params()
      • HyperOptSelector.get_params()
  • Module contents
• astrodata.ml.models package
  • Submodules
  • astrodata.ml.models.BaseMlModel module
    • BaseMlModel
      • BaseMlModel.clone()
      • BaseMlModel.fit()
      • BaseMlModel.get_metrics()
      • BaseMlModel.get_params()
      • BaseMlModel.get_scorer_metric()
      • BaseMlModel.load()
      • BaseMlModel.predict()
      • BaseMlModel.save()
      • BaseMlModel.score()
      • BaseMlModel.set_params()
  • astrodata.ml.models.PytorchModel module
  • astrodata.ml.models.SklearnModel module
    • SklearnModel
      • SklearnModel.fit()
      • SklearnModel.get_loss_history()
      • SklearnModel.get_loss_history_metrics()
      • SklearnModel.get_metrics()
      • SklearnModel.get_params()
      • SklearnModel.get_scorer_metric()
      • SklearnModel.has_loss_history
      • SklearnModel.load()
      • SklearnModel.predict()
      • SklearnModel.predict_proba()
      • SklearnModel.save()
      • SklearnModel.score()
      • SklearnModel.set_params()
  • astrodata.ml.models.TensorflowModel module
  • astrodata.ml.models.XGBoostModel module
    • XGBoostModel
      • XGBoostModel.fit()
      • XGBoostModel.get_loss_history()
      • XGBoostModel.get_loss_history_metrics()
      • XGBoostModel.get_metrics()
      • XGBoostModel.get_params()
      • XGBoostModel.get_scorer_metric()
      • XGBoostModel.has_loss_history
      • XGBoostModel.load()
      • XGBoostModel.predict()
      • XGBoostModel.save()
      • XGBoostModel.score()
      • XGBoostModel.set_params()
  • Module contents

Module contents