File tree 1 file changed +4
-4
lines changed
src/test/java/com/thealgorithms/machinelearning 1 file changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -11,7 +11,7 @@ class LinearRegressionTest {
11
11
void testLinearRegression () {
12
12
ArrayList <Double > dependentX = new ArrayList <>();
13
13
ArrayList <Double > independentY = new ArrayList <>();
14
-
14
+
15
15
dependentX .add (1.0 );
16
16
independentY .add (2.0 );
17
17
dependentX .add (2.0 );
@@ -22,14 +22,14 @@ void testLinearRegression() {
22
22
independentY .add (5.0 );
23
23
dependentX .add (5.0 );
24
24
independentY .add (6.0 );
25
-
25
+
26
26
// Create LinearRegression object
27
27
LinearRegression lr = new LinearRegression (dependentX , independentY );
28
-
28
+
29
29
// Check the slope (m) and intercept (c)
30
30
assertEquals (1.0 , lr .getM (), 0.001 );
31
31
assertEquals (1.0 , lr .getC (), 0.001 );
32
-
32
+
33
33
// Check prediction for X = 6
34
34
double predictedY = lr .predictForX (6.0 );
35
35
assertEquals (7.0 , predictedY , 0.001 );
You can’t perform that action at this time.
0 commit comments