preprocessor

type Image struct {
	// contains filtered or unexported fields
}

type ImageColor string

type ImageConfig struct {
	ColorMode     ImageColor
	ResizeX       int
	ResizeY       int
	Interpolation resize.InterpolationFunction
}

type ProcessedImage struct {
	Color ImageColor
	Image image.Image
}

type Processor struct {
	Name       string
	LineOffset int
	// TODO: find a nicer way to capture multiple columns in a row
	DataLength  int
	RequiresFit bool
	// contains filtered or unexported fields
}

type ProcessorConfig struct {
	CacheDir    string
	LineOffset  int
	DataLength  int
	RequiresFit bool
	Divisor     *RegressionDivisor
	Tokenizer   *Tokenizer
	Image       *Image
	Reader      func(column []string) interface{}
	Converter   func(column interface{}) (*tf.Tensor, error)
}

type RegressionDivisor struct {
	// contains filtered or unexported fields
}

type Tokenizer struct {
	// contains filtered or unexported fields
}

type TokenizerConfig struct {
	IsCategoryTokenizer bool
	Filters             string
	DisableFiltering    bool
}