@@ -131,7 +131,7 @@ discard block |
||
131 | 131 | |
132 | 132 | /** |
133 | 133 | * @param array $records |
134 | - * @return DecisionTreeLeaf[] |
|
134 | + * @return null|DecisionTreeLeaf |
|
135 | 135 | */ |
136 | 136 | protected function getBestSplit($records) |
137 | 137 | { |
@@ -254,7 +254,7 @@ discard block |
||
254 | 254 | |
255 | 255 | /** |
256 | 256 | * @param array $sample |
257 | - * @return mixed |
|
257 | + * @return string |
|
258 | 258 | */ |
259 | 259 | protected function predictSample(array $sample) |
260 | 260 | { |