Hi, Could you please decrease your step size to 0.1, and also try 0.01? You could also try running L-BFGS, which doesn't have step size tuning, to get better results.
Best, Burak On Tue, Jul 21, 2015 at 2:59 AM, Naveen <nav...@formcept.com> wrote: > Hi , > > I am trying to use LinearRegressionWithSGD on Million Song Data Set and my > model returns NaN's as weights and 0.0 as the intercept. What might be the > issue for the error ? I am using Spark 1.40 in standalone mode. > > Below is my model: > > val numIterations = 100 > val stepSize = 1.0 > val regParam = 0.01 > val regType = "L2" > val algorithm = new LinearRegressionWithSGD() > > algorithm.optimizer.setNumIterations(numIterations).setStepSize(stepSize).setRegParam(regParam) > val model = algorithm.run(parsedTrainData) > > Regards, > Naveen > > --------------------------------------------------------------------- > To unsubscribe, e-mail: user-unsubscr...@spark.apache.org > For additional commands, e-mail: user-h...@spark.apache.org > >
