Extra assertEqual that takes double arguments
Refactored StatefulLinearRegression

- Yields a LinearRegressionModel for each keyed point, extended to
include r2
- Uses SharedLinearRegressionModel for common parameters derived from
features
- Updated tests appropriately
Removed unused import