Skip to main content
Version: v2.1.0

CSV example

from featurestore import Client, CSVFile

# Initialise feature store client
client = Client("url")
client.auth.login()

# Set project specifics
project = client.projects.create("demo")

# Create the csv source
csv = CSVFile("s3a://h2o-datasets/taxi_small.csv")
csv_schema = client.extract_schema_from_source(csv)

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

# Ingest to cache
my_feature_set.ingest(csv)

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