File tree 1 file changed +6
-6
lines changed
1 file changed +6
-6
lines changed Original file line number Diff line number Diff line change @@ -39,13 +39,13 @@ def xgboost(
39
39
40
40
def main () -> None :
41
41
"""
42
- >>> main()
43
- Mean Absolute Error : 0.30957163379906033
44
- Mean Square Error : 0.22611560196662744
45
-
46
42
The URL for this algorithm
47
43
https://xgboost.readthedocs.io/en/stable/
48
44
California house price dataset is used to demonstrate the algorithm.
45
+
46
+ Expected error values:
47
+ Mean Absolute Error: 0.30957163379906033
48
+ Mean Square Error: 0.22611560196662744
49
49
"""
50
50
# Load California house price dataset
51
51
california = fetch_california_housing ()
@@ -55,8 +55,8 @@ def main() -> None:
55
55
)
56
56
predictions = xgboost (x_train , y_train , x_test )
57
57
# Error printing
58
- print (f"Mean Absolute Error : { mean_absolute_error (y_test , predictions )} " )
59
- print (f"Mean Square Error : { mean_squared_error (y_test , predictions )} " )
58
+ print (f"Mean Absolute Error: { mean_absolute_error (y_test , predictions )} " )
59
+ print (f"Mean Square Error: { mean_squared_error (y_test , predictions )} " )
60
60
61
61
62
62
if __name__ == "__main__" :
You can’t perform that action at this time.
0 commit comments