@dataclass
class PreprocessedFeatureVector(BaseDataClass):
float_features: torch.Tensor
sequence_features: Optional[SequenceFeatures] = None
// Experimental: sticking this here instead of putting it in float_features
// because a lot of places derive the shape of float_features from
// normalization parameters.
time_since_first: Optional[torch.Tensor] = None
After Change
@dataclass
class PreprocessedFeatureVector(BaseDataClass):
float_features: torch.Tensor
id_list_features: IdListFeatures = dataclasses.field(default_factory=dict)
// Experimental: sticking this here instead of putting it in float_features
// because a lot of places derive the shape of float_features from
// normalization parameters.
time_since_first: Optional[torch.Tensor] = None
