Skip to main content
Version: v3.0.0

MongoDb example

import h2o_featurestore
from h2o_featurestore import MongoDbCollection

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

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

# Create the MongoDB source
mongo_db_source = MongoDbCollection("mongodb+srv://some_cluster.mongodb.net/test", database="sample_guides", collection="planets")
schema = workspace.extract_schema_from_source(mongo_db_source)

# Register the feature set
my_feature_set = workspace.feature_sets.register(schema, "feature_set_name", primary_key="_id")

# Quick look on the data
my_feature_set.get_preview()
Feedback