Python API

此頁麵提供指向Databricks特性庫Python API文檔的鏈接,以及關於Python客戶端包“Databricks - Feature - Store”的信息。

Python API參考

使用表中的鏈接下載或顯示Feature Store Python API參考。要確定Databricks Runtime ML版本的預安裝版本,請參見Feature Store兼容性矩陣

版本

下載

在線API參考

V0.3.6及以上版本

Feature Store Python API 0.7.0參考PDF

在線API參考

V0.3.5及以下版本

Feature Store Python API 0.3.5參考PDF

在線API參考不可用

Python包

Databricks特性庫api可通過Python客戶端包“Databricks - Feature - Store”獲得。客戶端處於可用狀態PyPI並預先安裝在Databricks機器學習運行時中。有關哪個運行時包含哪個客戶端版本的參考,請參閱特性存儲兼容性矩陣

在Databricks運行時中安裝客戶端:

皮普安裝-功能-商店

在本地Python環境中安裝客戶端。

皮普安裝-功能-商店

支持場景

在Databricks上,包括Databricks運行時和Databricks運行時for Machine Learning,您可以:

  • 創建、讀取和寫入特性表。

  • 根據特征數據對模型進行訓練和評分。

  • 發布特性表到網上商店實時服務。

在本地環境或Databricks外部環境中,您可以:

  • 使用本地IDE支持開發代碼。

  • 使用模擬框架進行單元測試。

  • 編寫要在Databricks上運行的集成測試。

限製

客戶端庫隻能在Databricks上運行,包括Databricks Runtime和Databricks Runtime for Machine Learning。它不支持從本地環境調用Feature Store api,也不支持從Databricks以外的環境調用。

使用Feature Store客戶機進行單元測試

您可以在本地安裝功能庫客戶機,以幫助運行單元測試。

例如,驗證一個方法update_customer_features正確的電話FeatureStoreClient.write_table,你可以這樣寫:

unittest.mock進口MagicMock補丁my_feature_update_module進口update_customer_featuresdatabricks.feature_store進口FeatureStoreClient@patch對象FeatureStoreClient“write_table”@patch“my_feature_update_module.compute_customer_features”deftest_somethingcompute_customer_featuresmock_write_table):customer_features_dfMagicMock()compute_customer_featuresreturn_valuecustomer_features_dfupdate_customer_features()#測試功能mock_write_tableassert_called_once_with的名字“recommender_system.customer_features”dfcustomer_features_df模式“合並”

使用Feature Store客戶機進行集成測試

您可以在Databricks上使用Feature Store客戶端運行集成測試。詳情請參見開發人員工具和指導:使用CI/CD

在集成開發環境(IDE)中使用Feature Store客戶機

您可以將Feature Store客戶端與IDE一起使用,使用Databricks進行軟件開發。詳情請參見使用Databricks的IDE