php-ai / php-ml

src/Phpml/Classification/DecisionTree.php

Doc Comments +2 added lines, -4 removed lines

@@ -184,7 +184,7 @@ 
 
     /**
      * @param array $records
-     * @return DecisionTreeLeaf[]
+     * @return null|DecisionTreeLeaf
      */
     protected function getBestSplit($records)
     {
@@ -377,7 +377,7 @@ 
     /**
      * Used to set predefined features to consider while deciding which column to use for a split
      *
-     * @param array $selectedFeatures
+     * @param integer[] $selectedFeatures
      */
     protected function setSelectedFeatures(array $selectedFeatures)
     {
@@ -415,7 +415,6 @@ 
      * each column in the given dataset. The importance values are
      * normalized and their total makes 1.<br/>
      *
-     * @param array $labels
      * @return array
      */
     public function getFeatureImportances()
@@ -455,7 +454,6 @@ 
      *
      * @param int $column
      * @param DecisionTreeLeaf
-     * @param array $collected
      *
      * @return array
      */

src/Phpml/Classification/Linear/DecisionStump.php

Doc Comments +2 added lines, -2 removed lines

@@ -158,8 +158,8 @@
      * @param string $operator
      * @param array $values
      * @param array $targets
-     * @param mixed $leftLabel
-     * @param mixed $rightLabel
+     * @param string $leftLabel
+     * @param string $rightLabel
      */
     protected function calculateErrorRate(float $threshold, string $operator, array $values, array $targets, $leftLabel, $rightLabel)
     {