Create setup.py so we can have algos used as a library in other projects

algorithms/gk.py → fuzzy_clust_algos/gk.py

@@ -71,12 +71,15 @@ def next_u(self, d):
         return denominator_
 
     def predict(self, z):
+        u = self.predict_proba(z)
+        return np.argmax(u, axis=0)
+
+    def predict_proba(self, z):
         if len(z.shape) == 1:
             z = np.expand_dims(z, axis=0)
 
         dist = self._distance(z, self.centers, self.f)
         if len(dist.shape) == 1:
             dist = np.expand_dims(dist, axis=0)
 
-        u = self.next_u(dist)
-        return np.argmax(u, axis=0)
+        return self.next_u(dist)

setup.py

@@ -0,0 +1,14 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+from setuptools import setup, find_packages
+
+
+setup(name='fuzzy_clust_algos',
+      version="1.0",
+      description='Fuzzy Clustering Algorithms.',
+      packages=find_packages(),
+      install_requires=[
+          'numpy',
+          'scipy',
+      ],
+      )

ui.py

@@ -8,9 +8,9 @@
 from PyQt5.QtCore import Qt
 from PyQt5.QtWidgets import QFileSystemModel
 
-from algorithms.fcm import FCM
-from algorithms.gk import GK
 from file_path_manager import FilePathManager
+from fuzzy_clust_algos.fcm import FCM
+from fuzzy_clust_algos.gk import GK
 
 FormClass = uic.loadUiType("ui.ui")[0]