On a single axes, plot y1 against x with label "train" and y2 against x with label "test", then display the legend. Return the Axes.
y1
x
"train"
y2
"test"
solve(...)