Update comments, fix KMeans estimator

Author: rushter

mla/ensemble/random_forest.py

@@ -8,7 +8,7 @@
 
 class RandomForest(BaseEstimator):
     def __init__(self, n_estimators=10, max_features=None, min_samples_split=10, max_depth=None, criterion=None):
-        """
+        """Base class for RandomForest.
 
         Parameters
         ----------
@@ -42,6 +42,9 @@ def _train(self):
             tree.train(self.X, self.y, max_features=self.max_features, min_samples_split=self.min_samples_split,
                        max_depth=self.max_depth)
 
+    def _predict(self, X=None):
+        raise NotImplementedError()
+
 
 class RandomForestClassifier(RandomForest):
     def __init__(self, n_estimators=10, max_features=None, min_samples_split=10, max_depth=None, criterion='entropy'):

mla/ensemble/tree.py

@@ -82,6 +82,8 @@ def train(self, X, target, max_features=None, min_samples_split=10, max_depth=No
             Maximum depth of the tree.
         minimum_gain : float, default 0.01
             Minimum gain required for splitting.
+        loss : function, default None
+            Loss function for gradient boosting.
         """
 
         if not isinstance(target, dict):

mla/kmeans.py

@@ -59,7 +59,7 @@ def _initialize_cetroids(self, init):
         else:
             raise ValueError('Unknown type of init parameter')
 
-    def predict(self):
+    def _predict(self, X=None):
         """Perform the clustering on the dataset."""
         self._initialize_cetroids(self.init)
         centroids = self.centroids