Python SimpleLinearRegression Examples

Example #1

File: test_simple_linear_regression.py Project: dlosardo/machine_learning

class TestSimpleLinearRegression(object):

    def setup(self):
        print ("setup() before any methods in this class")
        self.hypo = SimpleLinearRegression()

    """
    Tests
    """
    def test_simple_linear_regression_instance(self):
        assert_is_instance(self.hypo, SimpleLinearRegression)

    def test_initialize_parameters_correctly_sets_parameter_estimates(self):
        self.hypo.initialize_parameters(PARAM_DICT)
        assert_equals(self.hypo.intercept.value, INTERCEPT_VALUE)
        assert_equals(self.hypo.slope.value, SLOPE_VALUE)

    def test_update_parameters_raises_error_without_initialization(self):
        assert_raises(ParameterValuesNotInitialized, self.hypo.update_parameters,
            PARAM_ARRAY)

Example #2

File: test_simple_linear_regression.py Project: dlosardo/machine_learning

 def setup(self):
     print ("setup() before any methods in this class")
     self.hypo = SimpleLinearRegression()