class BinaryInputFeatureConfig(BaseInputFeatureConfig): """BinaryInputFeature is a dataclass that configures the parameters used for a binary input feature.""" preprocessing: BasePreprocessingConfig = PreprocessingDataclassField(feature_type=BINARY) encoder: Optional[str] = schema_utils.StringOptions( list(get_encoder_classes(BINARY).keys()), default="passthrough", description="Encoder to use for this binary feature.", )
class DateInputFeatureConfig(BaseInputFeatureConfig): """DateInputFeature is a dataclass that configures the parameters used for a date input feature.""" preprocessing: BasePreprocessingConfig = PreprocessingDataclassField(feature_type=DATE) encoder: Optional[str] = schema_utils.StringOptions( list(get_encoder_classes(DATE).keys()), default="embed", description="Encoder to use for this date feature.", )
class CategoryInputFeatureConfig(BaseInputFeatureConfig): """CategoryInputFeature is a dataclass that configures the parameters used for a category input feature.""" preprocessing: BasePreprocessingConfig = PreprocessingDataclassField( feature_type=CATEGORY) encoder: Optional[str] = schema_utils.StringOptions( list(get_encoder_classes(CATEGORY).keys()), default="dense", description="Encoder to use for this category feature.", )
class ImageInputFeatureConfig(BaseInputFeatureConfig): """ImageInputFeatureConfig is a dataclass that configures the parameters used for an image input feature.""" preprocessing: BasePreprocessingConfig = PreprocessingDataclassField( feature_type=IMAGE) encoder: Optional[str] = schema_utils.StringOptions( list(get_encoder_classes(IMAGE).keys()), default="stacked_cnn", description="Encoder to use for this image feature.", )
class VectorInputFeatureConfig(BaseInputFeatureConfig): """VectorInputFeatureConfig is a dataclass that configures the parameters used for a vector input feature.""" preprocessing: BasePreprocessingConfig = PreprocessingDataclassField( feature_type=VECTOR) encoder: Optional[str] = schema_utils.StringOptions( list(get_encoder_classes(VECTOR).keys()), default="dense", description="Encoder to use for this vector feature.", )
class TextInputFeatureConfig(BaseInputFeatureConfig): """TextInputFeatureConfig is a dataclass that configures the parameters used for a text input feature.""" preprocessing: BasePreprocessingConfig = PreprocessingDataclassField( feature_type=TEXT) encoder: Optional[str] = schema_utils.StringOptions( list(get_encoder_classes(TEXT).keys()), default="parallel_cnn", description="Encoder to use for this text feature.", )
class NumberInputFeatureConfig(BaseInputFeatureConfig): """NumberInputFeature is a dataclass that configures the parameters used for a number input feature.""" preprocessing: BasePreprocessingConfig = PreprocessingDataclassField( feature_type=NUMBER) encoder: Optional[str] = schema_utils.StringOptions( list(get_encoder_classes(NUMBER).keys()), default="passthrough", description="Encoder to use for this number feature.", )
class AudioInputFeatureConfig(BaseInputFeatureConfig): """AudioFeatureInputFeature is a dataclass that configures the parameters used for an audio input feature.""" preprocessing: BasePreprocessingConfig = PreprocessingDataclassField( feature_type=AUDIO) encoder: Optional[str] = schema_utils.StringOptions( list(get_encoder_classes(AUDIO).keys()), default="parallel_cnn", description="Encoder to use for this audio feature.", )