I want to do it for more sophisticated ML algorithms too, but first started with simple linear regression.