ludwig-ai--ludwig
wehub-resource-sync
593b94c120
pytest / Unit Tests (push) Has been cancelled
pytest / Integration (integration_tests_a) (push) Has been cancelled
pytest / Integration (integration_tests_b) (push) Has been cancelled
pytest / Integration (integration_tests_c) (push) Has been cancelled
pytest / Integration (integration_tests_d) (push) Has been cancelled
pytest / Integration (integration_tests_e) (push) Has been cancelled
pytest / Integration (integration_tests_f) (push) Has been cancelled
pytest / Integration (integration_tests_g) (push) Has been cancelled
pytest / Integration (integration_tests_h) (push) Has been cancelled
pytest / Integration (integration_tests_i) (push) Has been cancelled
pytest / Integration (integration_tests_j) (push) Has been cancelled
pytest / Distributed (distributed_a) (push) Has been cancelled
pytest / Distributed (distributed_b) (push) Has been cancelled
pytest / Distributed (distributed_c) (push) Has been cancelled
pytest / Distributed (distributed_d) (push) Has been cancelled
pytest / Distributed (distributed_e) (push) Has been cancelled
pytest / Distributed (distributed_f) (push) Has been cancelled
pytest / Minimal Install (push) Has been cancelled
pytest / Event File (push) Has been cancelled
pytest (slow) / py-slow (push) Has been cancelled
Publish JSON Schema / publish-schema (push) Has been cancelled
33 行
1.1 KiB
Python
33 行
1.1 KiB
Python
"""Ludwig AutoML — automated config generation and training.
|
|
|
|
Public API
|
|
----------
|
|
auto_train(dataset, target, time_limit_s)
|
|
End-to-end: run dataset quality checks, sample configs from the YAML-driven
|
|
search space, train each, and return the best model.
|
|
|
|
create_auto_config(dataset, target)
|
|
Return a single sampled Ludwig config dict without training.
|
|
|
|
train_with_config(dataset, config, output_dir)
|
|
Train one Ludwig config and return the LudwigModel.
|
|
|
|
cli_init_config(argv)
|
|
CLI entry-point for `ludwig init_config`.
|
|
|
|
Config generation pipeline
|
|
--------------------------
|
|
The search space is defined by YAML files in ``ludwig/automl/search_space/``.
|
|
``config_enumerator`` builds the full combination lattice, ``config_sampler``
|
|
draws a random subset, and ``config_validator`` rejects invalid combinations
|
|
before any training starts. ``target_detection`` infers the task type and
|
|
target column when they are not explicitly specified.
|
|
"""
|
|
|
|
from ludwig.automl.automl import (
|
|
auto_train, # noqa: F401
|
|
cli_init_config, # noqa: F401
|
|
create_auto_config, # noqa: F401
|
|
train_with_config, # noqa: F401
|
|
)
|