diff --git a/src/Phpml/Classifier/Classifier.php b/src/Phpml/Classifier/Classifier.php index ea2bbf1..1d2362e 100644 --- a/src/Phpml/Classifier/Classifier.php +++ b/src/Phpml/Classifier/Classifier.php @@ -1,4 +1,5 @@ k = $k; + $this->features = []; + $this->labels = []; + } + + + /** + * @param array $features + * @param array $labels + */ + public function train(array $features, array $labels) + { + $this->features = $features; + $this->labels = $labels; + } + + /** + * @param mixed $feature + * @return mixed + */ + public function predict($feature) + { + + } + +} diff --git a/src/Phpml/Classifier/NaiveBayes.php b/src/Phpml/Classifier/NaiveBayes.php index 7207409..8e31eff 100644 --- a/src/Phpml/Classifier/NaiveBayes.php +++ b/src/Phpml/Classifier/NaiveBayes.php @@ -1,10 +1,27 @@