Feature Store

The public API allows you to:

• list feature groups dataikuapi.dss.feature_store.DSSFeatureStore.list_feature_groups()

• check if a dataset is a feature group dataikuapi.dss.dataset.DSSDatasetSettings.is_feature_group()

• set/unset a dataset as a feature group: dataikuapi.dss.dataset.DSSDatasetSettings.set_feature_group()

See Feature Store for more information.

Listing feature groups

import dataiku

# if using API from inside DSS
client = dataiku.api_client()

feature_store = client.get_feature_store()

feature_groups = feature_store.list_feature_groups()

for feature_group in feature_groups:
    print("{}".format(feature_group.id))

Note

This will only display feature groups of projects on which the user has at least read permission

Note

Because of indexing latency, you have have to wait a few seconds before newly defined feature groups are visible

(Un)setting a dataset as a Feature Group

import dataiku

# if using API from inside DSS
client = dataiku.api_client()

project = client.get_project('PROJECT_ID')

ds = project.get_dataset('DATASET_ID')

ds_settings = ds.get_settings()

# pass False to undefine as Feature Group
ds_settings.set_feature_group(True)

ds_settings.save()

API Reference

class dataikuapi.dss.feature_store.DSSFeatureStore(client)

list_feature_groups()

Get a list of feature groups on which the user has at least read permissions

Returns

list of feature groups

Return type

list of dataikuapi.feature_store.DSSFeatureGroupListItem

class dataikuapi.dss.feature_store.DSSFeatureGroupListItem(client, project_key, name)

property id

get_as_dataset()

Gets the feature group as a dataset

Returns

a handle on the dataset

Return type

dataikuapi.dss.dataset.DSSDataset