AutoMAX

API Reference

  • API Reference
    • src.core
    • AutoMAX
      • AutoMAX
        • AutoMAX.__init__()
        • AutoMAX.train()
        • AutoMAX.optimize()
    • AutoMAXConfigration
      • AutoMAXConfigration
    • autopartial
      • autopartial
        • autopartial.__call__()
    • src.core.helpers
      • parse_hyperparameters_from_dict()
    • src.auto_trainer
      • main()
      • apply_cli_overrides()
      • set_seed()
    • src.auto_gnn

Get Started

  • Quick Start
    • Step 1 — Install
    • Step 2 — Write a config
    • Step 3 — Run AutoMAX
    • Entry Points
  • Installation
    • Requirements
    • From Source
    • Core Dependencies
    • Verifying the Install
  • Configuration Reference
    • dataset
    • model
    • metrics
    • training
      • Experiment tracking
      • Core training parameters
      • Loss and optimizer
      • Learning rate schedule
      • Checkpointing
    • automax

AutoMAX Search

  • AutoMAX Search
    • How It Works
    • Resuming a Search
    • What Gets Tuned
    • Output
  • Loss / Optimizer Pairs
    • Choosing a Pair
  • Hyperparameter Definition Format
    • Scalar Constant
    • Search Space Dict
      • Sub-fields
    • Examples
  • CLI Overrides
    • Reference

Recipes

  • Recipes
    • AUROC — AUCMLoss + PESG
      • Config
      • Run
      • Hyperparameter search space
    • AUPRC — APLoss + SOAP
      • Config
      • Run
    • One-way pAUC — pAUCLoss (1w) + SOPAs
      • Config
      • Run
    • Two-way pAUC — pAUCLoss (2w) + SOTAs
      • Config
      • Run
    • Two-way pAUC CVaR — tpAUC_CVaR_loss + STACO
      • Config
      • Run
AutoMAX
  • Search

© Copyright 2024, Optimization-AI Lab.

Built with Sphinx using a theme provided by Read the Docs.