Package ml.shifu.guagua.example.lnr
Linear regression example by using batch gradient descent.
-
Class Summary Class Description LinearRegressionContants LinearRegressionMaster LinearRegressionMasterdefines logic to update global linear regression model.LinearRegressionOutput LinearRegressionOutputis used to write the final model output to file system.LinearRegressionParams A model class to store linear regression weight on first iteration by usingLinearRegressionParams.parameters, while in other iterationsLinearRegressionParams.parametersis used to store gradients.LinearRegressionWorker LinearRegressionWorkerdefines logic to accumulate local linear regression gradients.