Skip to main content
Version: v3.0.0

Parquet example

import h2o_featurestore
from h2o_featurestore import ParquetFile

# Initialise feature store client
client = h2o_featurestore.login()

# Set workspace specifics
workspace = client.workspaces.create("demo")

# Create the parquet source
parquet = ParquetFile("<path to parquet file>")
parquet_schema = workspace.extract_schema_from_source(parquet)

# Register the feature set
my_feature_set = workspace.feature_sets.register(parquet_schema, "feature_set_name", primary_key=["key_name"])

# Ingest to cache
my_feature_set.ingest(parquet)

# Retrieve feature set
ref = my_feature_set.retrieve()
ref.download()
Feedback