@@ -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 | { |