Made changes as per changes in load_data. Also added feature of saving weights and plot only if the user wants to.
Made changes as per changes in load_data.
Also added feature of saving weights and plot only if the user wants to.

sumanth-nirmal/self-driving-car-predict-steering,sumanth-nirmal/self-driving-car-predict-steering