@@ -37,12 +37,11 @@ def test_predict_hash
3737 pred = booster . predict ( { "x3" => 9.0 , "x2" => 7.2 , "x1" => 1.2 , "x0" => 3.7 } )
3838 assert_in_delta 0.9823112229173586 , pred
3939
40- pred = booster . predict (
41- [
40+ pred =
41+ booster . predict ( [
4242 { "x3" => 9.0 , "x2" => 7.2 , "x1" => 1.2 , "x0" => 3.7 } ,
43- { "x3" => 0.0 , "x2" => 7.9 , "x1" => 0.5 , "x0" => 7.5 } ,
44- ]
45- )
43+ { "x3" => 0.0 , "x2" => 7.9 , "x1" => 0.5 , "x0" => 7.5 }
44+ ] )
4645 assert_elements_in_delta [ 0.9823112229173586 , 0.9583143724610858 ] , pred . first ( 2 )
4746
4847 assert_raises ( KeyError ) do
@@ -54,7 +53,7 @@ def test_predict_daru
5453 x_test =
5554 Daru ::DataFrame . new ( [
5655 { "x3" => 9.0 , "x2" => 7.2 , "x1" => 1.2 , "x0" => 3.7 } ,
57- { "x3" => 0.0 , "x2" => 7.9 , "x1" => 0.5 , "x0" => 7.5 } ,
56+ { "x3" => 0.0 , "x2" => 7.9 , "x1" => 0.5 , "x0" => 7.5 }
5857 ] )
5958 pred = booster . predict ( x_test )
6059 assert_elements_in_delta [ 0.9823112229173586 , 0.9583143724610858 ] , pred . first ( 2 )
@@ -65,7 +64,7 @@ def test_predict_rover
6564 x_test =
6665 Rover ::DataFrame . new ( [
6766 { "x3" => 9.0 , "x2" => 7.2 , "x1" => 1.2 , "x0" => 3.7 } ,
68- { "x3" => 0.0 , "x2" => 7.9 , "x1" => 0.5 , "x0" => 7.5 } ,
67+ { "x3" => 0.0 , "x2" => 7.9 , "x1" => 0.5 , "x0" => 7.5 }
6968 ] )
7069 pred = booster . predict ( x_test )
7170 assert_elements_in_delta [ 0.9823112229173586 , 0.9583143724610858 ] , pred . first ( 2 )
0 commit comments