Merge pull request #81 from ealcobaca/complexity-fts

New complexity metafeatures

Makefile

@@ -37,7 +37,7 @@ pypi: clean ## Send the package to pypi.
 	twine upload dist/*
 
 install-dev: ## Install pymfe for developers using pip.
-	pip install -e .
+	pip install -U -e .
 	pip install -U -r requirements.txt
 	pip install -U -r requirements-dev.txt
 	pip install -U -r requirements-docs.txt

pymfe/_utils.py

@@ -1,15 +1,21 @@
 """Keeps generic utility functions."""
+import typing as t
+
 import numpy as np
 
 
-def calc_cls_inds(y: np.ndarray, classes: np.ndarray) -> np.ndarray:
+def calc_cls_inds(y: np.ndarray,
+                  classes: t.Optional[np.ndarray] = None) -> np.ndarray:
     """Compute the ``cls_inds`` variable.
 
     The ``cls_inds`` variable is a boolean array which marks with
     True value whether the instance belongs to each class. Each
     distinct class is represented by a row, and each instance is
     represented by a column.
     """
+    if classes is None:
+        classes = np.unique(y)
+
     cls_inds = np.array([np.equal(y, cur_cls) for cur_cls in classes],
                         dtype=bool)