emmi_inference.models.pipelines.default_tensor_precollator

WARNING This file is a 1:1 duplicate from the tutorial folder.

It is here to avoid installation of a tutorial as a package and keep it isolated.

Classes

DefaultTensorPreCollator

Create a tensor with a fixed dummy value, with a specified size.

Module Contents

class emmi_inference.models.pipelines.default_tensor_precollator.DefaultTensorPreCollator(item_key_name, feature_dim, size=None, matching_item_key=None, default_value=0.0)

Bases: ksuit.data.pipeline.SampleProcessor

Create a tensor with a fixed dummy value, with a specified size.

Initializes the DefaulTensorPreCollator.

Parameters:

• item_key_name (str) – _description_

• default_value (float) – _description_

• feature_dim (int) – _description_

• size (int) – _description_

• matching_item_key (str) – _description_

item_key_name

feature_dim

size = None

matching_item_key = None

default_value = 0.0