creators

type BinDiscretizer struct {
	// Columns to discretize
	Columns []string

	// NBins is the number of bins to create
	NBins int

	// Strategy for creating bins:
	// - "uniform": equal width bins
	// - "quantile": equal frequency bins
	// - "kmeans": k-means based bins (simplified)
	Strategy string

	// Encode specifies how to encode the bins:
	// - "ordinal": 0, 1, 2, ...
	// - "onehot": create binary columns for each bin
	Encode string
	// contains filtered or unexported fields
}

type InteractionFeatures struct {
	// Columns to create interactions from. If nil, use all numeric columns.
	Columns []string
	// contains filtered or unexported fields
}

type PolynomialFeatures struct {
	// Degree of the polynomial features
	Degree int

	// IncludeBias adds a constant feature (all 1s)
	IncludeBias bool

	// InteractionOnly only creates interaction features (no powers)
	InteractionOnly bool

	// Columns to use. If nil, use all numeric columns.
	Columns []string
	// contains filtered or unexported fields
}