Registry / ai-ml / databricks-automl-runtime

databricks-automl-runtime

JSON →
library0.2.21pypypiunverified

The Databricks AutoML Runtime package provides utilities and custom transformers designed to integrate with Databricks AutoML, particularly for time series and other specialized machine learning tasks. It offers custom scikit-learn compatible transformers, hyperparameter tuning wrappers for models like Prophet and pmdarima, and MLflow logging integrations. The current version is 0.2.21, and it has a moderate release cadence, often addressing compatibility or bug fixes.

pip install databricks-automl-runtime
INSTALL
IMPORT
SIG · DATABRICKS-AUTOML-
D
databricks-automl-runtime
ai-mlpythonv0.2.21
Install
1.6s avg
Import
Disk
16MB
Pass rate
10/ 10
Env Coverage10 / 10
glibc
3.93.13
musl
3.93.13
Install & Compatibility
Where this runs
tested against v0.2.21 · 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
py 3.103.930 runs
installs and imports cleanly · install 0.0s · import 0.000s · 18.2MB
glibc
py 3.103.930 runs
installs and imports cleanly · install 1.6s · import 0.000s · 19MB
16MB installed
● package 16MB
Code
Verified usage

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

DateTransformer
from databricks.automl_runtime.sklearn.date_time_transformers import DateTransformer
from databricks.automl_runtime.sklearn.date_time_transformers import DateTransformer

This quickstart demonstrates how to use the `DateTransformer` to extract various date and time features from a datetime column in a Pandas DataFrame. This is a common preprocessing step for time series models.

import pandas as pd from automl_runtime.sklearn.date_time_transformers import DateTransformer # Create a sample DataFrame with a datetime column df = pd.DataFrame({ 'timestamp_col': pd.to_datetime(['2023-01-01', '2023-01-02', '2023-01-03']), 'value': [10, 12, 15] }) # Instantiate the DateTransformer # This transformer extracts date-related features like year, month, day, day_of_week, etc. date_transformer = DateTransformer( timestamp_col='timestamp_col', output_timestamp_col_name='datetime_features' ) # Fit and transform the DataFrame X_transformed = date_transformer.fit_transform(df) print("Original DataFrame:\n", df) print("\nTransformed DataFrame (first 5 columns):\n", X_transformed.iloc[:, :5])
Debug
Known issues
breakingThe library explicitly requires `hyperopt>=0.2.7`. Using older versions of hyperopt can lead to unexpected errors or crashes during hyperparameter tuning tasks, particularly when using `HyperoptEstimator` classes.
fix
Ensure `hyperopt` is updated: `pip install -U "hyperopt>=0.2.7"`
affects: <0.2.4
gotchaProphet and pmdarima are optional dependencies. If you intend to use `ProphetHyperoptEstimator` or `ArimaHyperoptEstimator`, you must install the respective packages separately using the extra syntax (e.g., `pip install databricks-automl-runtime[prophet]`). A `ModuleNotFoundError` will occur otherwise.
fix
Install with extras: `pip install databricks-automl-runtime[prophet]` or `pip install databricks-automl-runtime[pmdarima]`.
affects: All versions
gotchaVersions of the library before `0.2.4.1` (e.g., `0.2.3.1`, `0.2.4`) had known issues with Prophet hyper-parameter enumeration. This could lead to incorrect tuning outcomes or runtime errors when defining certain hyperparameter search spaces for Prophet models.
fix
Upgrade `databricks-automl-runtime` to version `0.2.4.1` or newer: `pip install -U databricks-automl-runtime`.
affects: <0.2.4.1
Upgrade
Version history
0.2.21latest on PyPI · released Feb 21, 2024
Audit
Dependencies
mlflowrequiredCore dependency for MLflow integration and experiment tracking.
scikit-learnrequiredProvides base classes and utilities for sklearn-compatible transformers.
pandasrequiredData manipulation for input/output data structures.
pyyamlrequiredUsed for configuration and serialization.
scipyrequiredScientific computing dependency.
numpyrequiredNumerical computing dependency.
hyperoptrequiredRequired for hyperparameter tuning functionalities, specifically HyperoptEstimators.
prophetoptionalOptional: Required for Prophet-based time series models and hyperparameter tuning.
pmdarimaoptionalOptional: Required for pmdarima-based ARIMA time series models and hyperparameter tuning.
Agent activity
20 hits · last 30 days
node
18
OpenAI (training)
1
Resources
databricks-automl-runtime — pip install databricks-automl-runtime · libregistry