diff --git a/src/Phpml/Classification/MLPClassifier.php b/src/Phpml/Classification/MLPClassifier.php
index bde49a2..7f9043b 100644
--- a/src/Phpml/Classification/MLPClassifier.php
+++ b/src/Phpml/Classification/MLPClassifier.php
@@ -11,7 +11,10 @@ class MLPClassifier extends MultilayerPerceptron implements Classifier
 {
 
     /**
-     * @param  mixed $target
+     * @param mixed $target
+     *
+     * @throws InvalidArgumentException
+     *
      * @return int
      */
     public function getTargetClass($target): int
diff --git a/src/Phpml/Exception/InvalidArgumentException.php b/src/Phpml/Exception/InvalidArgumentException.php
index 277aecd..b618d00 100644
--- a/src/Phpml/Exception/InvalidArgumentException.php
+++ b/src/Phpml/Exception/InvalidArgumentException.php
@@ -67,6 +67,8 @@ class InvalidArgumentException extends \Exception
     }
 
     /**
+     * @param mixed $target
+     *
      * @return InvalidArgumentException
      */
     public static function invalidTarget($target)
diff --git a/src/Phpml/NeuralNetwork/Network/MultilayerPerceptron.php b/src/Phpml/NeuralNetwork/Network/MultilayerPerceptron.php
index 2503774..b7364b2 100644
--- a/src/Phpml/NeuralNetwork/Network/MultilayerPerceptron.php
+++ b/src/Phpml/NeuralNetwork/Network/MultilayerPerceptron.php
@@ -114,6 +114,9 @@ abstract class MultilayerPerceptron extends LayeredNetwork implements Estimator,
     /**
      * @param array $samples
      * @param array $targets
+     * @param array $classes
+     *
+     * @throws InvalidArgumentException
      */
     public function partialTrain(array $samples, array $targets, array $classes = [])
     {