Building a Regression Model with Spark - Machine Learning with Spark - page 242

Here, we will use the matplotlib library to plot a graph of the RMSLE metric against
the number of iterations. We will use a log scale for the x axis to make the output easier to
visualize:
plot(params, metrics)
fig = matplotlib.pyplot.gcf()
pyplot.xscale('log')
Metrics for varying number of iterations
Step size
We will perform a similar analysis for step size in the following code:
params = [0.01, 0.025, 0.05, 0.1, 1.0]
metrics = [evaluate(train_data, test_data, 10, param , 0.0,
'l2', False) for param in params]