MLKit icon indicating copy to clipboard operation
MLKit copied to clipboard

What should initial weights be?

Open junaidxabd opened this issue 4 years ago • 0 comments

I left the initial weights as the default values but it doesn't set the weights after that

        let initial_weights = Matrix<Float>(rows: 3, columns: 1, elements: [-100000.0, 1.0, 1.0])

        let weights = try! polynomialModel.train([timestamp,weekday], output: output_data, initialWeights: initial_weights, stepSize: Float(4e-12), tolerance: Float(1e9))

I've got 2 features like the Polynomial Regression Usage guide.

junaidxabd avatar Jun 16 '21 17:06 junaidxabd