SoftmaxRegression

@Serializable

@SerialName(value = "SoftmaxRegression")

data class SoftmaxRegression(val featureSize: Int, val numClasses: Int, val optimizer: OptimizerSpec = Sgd(), val biasOptimizer: OptimizerSpec = optimizer) : RegressionStatSpec<SoftmaxRegressionResult> (source)

Spec for SoftmaxRegressionStat: multinomial (K-way) logistic regression.

Constructors

constructor(featureSize: Int, numClasses: Int, optimizer: OptimizerSpec = Sgd(), biasOptimizer: OptimizerSpec = optimizer)(source)

Bias optimizer; defaults to optimizer.

Number of input features.

Number of classes; the input y must round to [0, numClasses).

Per-class weight-matrix optimizer; one instance is materialised per class.