imputers

type IterativeImputer struct {
	// MaxIter is the maximum number of imputation rounds
	MaxIter int

	// Tolerance for convergence
	Tolerance float64

	// InitialStrategy for first imputation ("mean" or "median")
	InitialStrategy string
	// contains filtered or unexported fields
}

type KNNImputer struct {
	// NNeighbors is the number of neighbors to use
	NNeighbors int

	// Weights specifies the weight function:
	// - "uniform": all neighbors have equal weight
	// - "distance": weight inversely proportional to distance
	Weights string
	// contains filtered or unexported fields
}

type SimpleImputer struct {
	// Columns to impute. If nil, all columns with nulls are imputed.
	Columns []string

	// Strategy for imputation:
	// - "mean": replace with mean (numeric only)
	// - "median": replace with median (numeric only)
	// - "most_frequent": replace with mode
	// - "constant": replace with FillValue
	Strategy string

	// FillValue is used when Strategy="constant"
	FillValue any
	// contains filtered or unexported fields
}