diff --git a/src/Phpml/Classification/NaiveBayes.php b/src/Phpml/Classification/NaiveBayes.php
deleted file mode 100644
index d7f5ea4..0000000
--- a/src/Phpml/Classification/NaiveBayes.php
+++ /dev/null
@@ -1,8 +0,0 @@
-<?php
-
-namespace Phpml\Classification;
-
-class NaiveBayes
-{
-
-}
diff --git a/src/Phpml/Classifier/Classifier.php b/src/Phpml/Classifier/Classifier.php
new file mode 100644
index 0000000..ea2bbf1
--- /dev/null
+++ b/src/Phpml/Classifier/Classifier.php
@@ -0,0 +1,20 @@
+<?php
+
+namespace Phpml\Classifier;
+
+interface Classifier
+{
+
+    /**
+     * @param array $features
+     * @param array $labels
+     */
+    public function train($features, $labels);
+
+    /**
+     * @param mixed $feature
+     * @return mixed
+     */
+    public function predict($feature);
+
+}
diff --git a/src/Phpml/Classifier/NaiveBayes.php b/src/Phpml/Classifier/NaiveBayes.php
new file mode 100644
index 0000000..78ac0ce
--- /dev/null
+++ b/src/Phpml/Classifier/NaiveBayes.php
@@ -0,0 +1,26 @@
+<?php
+
+namespace Phpml\Classifier;
+
+class NaiveBayes implements Classifier
+{
+
+    /**
+     * @param array $features
+     * @param array $labels
+     */
+    public function train($features, $labels)
+    {
+
+    }
+
+    /**
+     * @param mixed $feature
+     * @return mixed
+     */
+    public function predict($feature)
+    {
+
+    }
+
+}