Registry / azure / azureml-train-automl

azureml-train-automl

JSON →
library1.62.0pypypiunverified

The `azureml-train-automl` package is part of the Azure Machine Learning SDK v1, designed for automatically finding the best machine learning model and its parameters. It streamlines model selection, hyperparameter tuning, and feature engineering for various ML tasks like classification, regression, and forecasting. As of v1.62.0, Azure Machine Learning SDK v1 is deprecated with support ending on June 30, 2026. Users are strongly advised to migrate to Azure Machine Learning SDK v2 for continued support and new features.

pip install azureml-train-automl
INSTALL
IMPORT
SIG · AZUREML-TRAIN-AUTO
A
azureml-train-automl
azurepythonv1.62.0
Install
90.3s avg
Import
3650ms
Disk
1741MB
Pass rate
1/ 10
Env Coverage1 / 10
glibc
3.93.13
musl
3.93.13
Install & Compatibility
Where this runs
tested against v1.62.0 · pip install
no network on importno background threads
Install × environment matrix
Each cell = how many times install + import succeeded across repeated harness runs. Partial = flaky.
glibc = Debian/Ubuntu slim · musl = Alpine Linux
musl
glibc
py 3.10
✕ build_error
✓ 90.25s
py 3.11
✕ build_error
✕ timeout
py 3.12
✕ build_error
✕ build_error
py 3.13
✕ build_error
✕ build_error
py 3.9
✕ build_error
✕ timeout
1741MB installed
● package 1741MB
Code
Verified usage

Verified import paths — ran on the pinned version, not inferred.

AutoMLConfig
from azureml.train.automl import AutoMLConfig
from azureml.automl.core import AutoMLConfig
The `azureml.automl.core` package is intended for internal SDK use and not direct consumption.
Experiment
from azureml.core.experiment import Experiment
Workspace
from azureml.core import Workspace
Dataset
from azureml.core.dataset import Dataset

This quickstart demonstrates how to set up and submit an AutoML experiment using `AutoMLConfig`. It includes connecting to an Azure Machine Learning Workspace, loading sample data, and configuring basic AutoML settings for a classification task. Ensure your Azure ML Workspace is configured (via `config.json` or environment variables) and a compute target is available. For actual execution, remote AmlCompute is recommended over 'local' to avoid extensive local dependency conflicts.

import os import pandas as pd from azureml.core.workspace import Workspace from azureml.core.experiment import Experiment from azureml.core.dataset import Dataset from azureml.train.automl import AutoMLConfig # NOTE: Replace with your actual workspace details or ensure config.json is present try: ws = Workspace.from_config() print(f"Workspace loaded: {ws.name}") except Exception as e: print(f"Could not load workspace from config. Attempting environment variables. Error: {e}") subscription_id = os.environ.get("AZUREML_SUBSCRIPTION_ID", "YOUR_SUBSCRIPTION_ID") resource_group = os.environ.get("AZUREML_RESOURCE_GROUP", "YOUR_RESOURCE_GROUP") workspace_name = os.environ.get("AZUREML_WORKSPACE_NAME", "YOUR_WORKSPACE_NAME") if "YOUR_" in subscription_id + resource_group + workspace_name: raise ValueError("Please configure your Azure ML Workspace details via config.json or environment variables.") ws = Workspace.get(name=workspace_name, subscription_id=subscription_id, resource_group=resource_group) print(f"Workspace loaded from env: {ws.name}") experiment_name = "automl-quickstart-exp" experiment = Experiment(ws, experiment_name) # Load sample data (replace with your own Dataset registration or data path) data_url = "https://automlsamplenotebookdata.blob.core.windows.net/automl-sample-notebook-data/creditcard.csv" df = pd.read_csv(data_url) # For simplicity, create a dummy Dataset. In a real scenario, you'd register your data. # Or use a registered dataset: Dataset.get_by_name(ws, name='my_dataset') from azureml.data.tabulardataset import TabularDataset training_data = TabularDataset.from_pandas_dataframe(df, target=(ws.get_default_datastore(), 'automl_creditcard.csv')) # Configure AutoML automl_config = AutoMLConfig( task='classification', primary_metric='accuracy', training_data=training_data, label_column_name='Class', compute_target='local', experiment_timeout_minutes=15, max_concurrent_iterations=1, n_cross_validations=2, iterations=5, verbosity=logging.INFO ) # Submit the AutoML run # NOTE: 'local' compute target runs on the current environment, may require many local dependencies. # For remote compute, configure an AmlCompute target and specify it in AutoMLConfig. print("Submitting AutoML experiment...") # run = experiment.submit(automl_config, show_output=True) # print(f"AutoML experiment submitted: {run.id}") # print("NOTE: Uncomment the submit line and ensure compute target is configured for actual execution.") import logging # Ensure logging is imported for verbosity
Debug
Known issues
breakingAzure Machine Learning SDK v1, which `azureml-train-automl` is part of, is deprecated as of March 31, 2025. Support will end on June 30, 2026. Existing workflows will continue to operate but could be exposed to security risks or breaking changes.
fix
Migrate your ML workflows to Azure Machine Learning SDK v2. Refer to the official Azure ML documentation for migration guides.
affects: All v1 versions after 2025-03-31
gotchaUpgrading `azureml-train-automl` from versions prior to `1.0.76` can lead to partial installations and import failures due to internal dependency conflicts.
fix
Before upgrading, completely uninstall the old version using `pip uninstall azureml-train-automl`. Alternatively, after upgrading, run `pip install --upgrade azureml-train-automl` followed by `pip install --ignore-installed azureml-train-automl-client`.
affects: <1.0.76 to >=1.0.76
breakingSeveral AutoML algorithms for Regression (`FastLinearRegressor`, `OnlineGradientDescentRegressor`) and Classification (`AveragedPerceptronClassifier`) were deprecated and are no longer supported in versions `1.49.0` and above.
fix
If these specific algorithms are required, use `azureml-train-automl` v1.48.0 or below, or switch to alternative algorithms. Consider migrating to SDK v2 for modern algorithm support.
affects: >=1.49.0
gotchaDependency mismatches, especially with `pandas` and `scikit-learn`, are a common source of errors (e.g., `Module not found`, `ImportError`, `AttributeError`) due to strict version pinning in older SDK v1 releases.
fix
For SDK versions `>1.13.0`, ensure `pandas==0.25.1` and `scikit-learn==0.22.1`. Use `pip install --upgrade pandas==0.25.1` and `pip install --upgrade scikit-learn==0.22.1`. Always verify the required environment from your experiment's `conda.yml` for exact versions.
affects: All v1 versions, especially after 1.13.0
Upgrade
Version history
1.62.0latest on PyPI · released Feb 25, 2026
Audit
Dependencies
azureml-corerequiredRequired for workspace connection, experiment management, and core Azure ML functionalities.
Agent activity
59 hits · last 30 days
node
52
OpenAI (training)
1
Resources
azureml-train-automl — pip install azureml-train-automl · libregistry