change interfaces and add Estimator and Pipeline

src/Phpml/Classification/Classifier.php

@@ -4,18 +4,8 @@
 
 namespace Phpml\Classification;
 
-interface Classifier
-{
-    /**
-     * @param array $samples
-     * @param array $labels
-     */
-    public function train(array $samples, array $labels);
+use Phpml\Estimator;
 
-    /**
-     * @param array $samples
-     *
-     * @return mixed
-     */
-    public function predict(array $samples);
+interface Classifier extends Estimator
+{
 }

src/Phpml/Estimator.php

@@ -0,0 +1,21 @@
+<?php
+
+declare (strict_types = 1);
+
+namespace Phpml;
+
+interface Estimator
+{
+    /**
+     * @param array $samples
+     * @param array $targets
+     */
+    public function train(array $samples, array $targets);
+
+    /**
+     * @param array $samples
+     *
+     * @return mixed
+     */
+    public function predict(array $samples);
+}

src/Phpml/FeatureExtraction/TfIdfTransformer.php

@@ -4,6 +4,8 @@
 
 namespace Phpml\FeatureExtraction;
 
+use Phpml\Transformer;
+
 class TfIdfTransformer implements Transformer
 {
     /**

src/Phpml/FeatureExtraction/TokenCountVectorizer.php

@@ -5,6 +5,7 @@
 namespace Phpml\FeatureExtraction;
 
 use Phpml\Tokenization\Tokenizer;
+use Phpml\Transformer;
 
 class TokenCountVectorizer implements Transformer
 {

src/Phpml/Pipeline.php

@@ -0,0 +1,29 @@
+<?php
+
+declare (strict_types = 1);
+
+namespace Phpml\Pipeline;
+
+class Pipeline
+{
+    /**
+     * @var array
+     */
+    private $stages;
+
+    /**
+     * @param array $stages
+     */
+    public function __construct(array $stages)
+    {
+        $this->stages = $stages;
+    }
+
+    /**
+     * @param mixed $stage
+     */
+    public function addStage($stage)
+    {
+        $this->stages[] = $stage;
+    }
+}

src/Phpml/Regression/Regression.php

@@ -4,18 +4,8 @@
 
 namespace Phpml\Regression;
 
-interface Regression
-{
-    /**
-     * @param array $samples
-     * @param array $targets
-     */
-    public function train(array $samples, array $targets);
+use Phpml\Estimator;
 
-    /**
-     * @param array $samples
-     *
-     * @return mixed
-     */
-    public function predict(array $samples);
+interface Regression extends Estimator
+{
 }

src/Phpml/FeatureExtraction/Transformer.php → src/Phpml/Transformer.php

@@ -2,7 +2,7 @@
 
 declare (strict_types = 1);
 
-namespace Phpml\FeatureExtraction;
+namespace Phpml;
 
 interface Transformer
 {