Merge pull request #6 from ydataai/feat/package

feat: Package
ydataai · Oct 14, 2020 · ee122f3 · ee122f3
2 parents 1648865 + 78b650e
commit ee122f3
Show file tree

Hide file tree

Showing 17 changed files with 46 additions and 18 deletions.
diff --git a/examples/cgan_example.py b/examples/cgan_example.py
@@ -1,5 +1,4 @@
 from models import CGAN
-from preprocessing.credit_fraud import *
 
 from sklearn import cluster
 

diff --git a/models/__init__.py b/models/__init__.py
diff --git a/requirements.txt b/requirements.txt
@@ -1,7 +1,6 @@
 pandas==1.0.3
 numpy==1.17.4
 scikit-learn==0.22.2
-matplotlib
-easydict
-
-tensorflow==2.1.0
+matplotlib==3.3.2
+tensorflow==2.1.0
+easydict
diff --git a/setup.py b/setup.py
@@ -0,0 +1,31 @@
+from setuptools import setup, find_namespace_packages
+import os
+from pathlib import Path
+
+here = Path(__file__).parent.resolve()
+
+requirements = (here / "requirements.txt").read_text(encoding="utf8")
+long_description = (here / 'README.md').read_text(encoding='utf-8')
+
+VERSION = os.getenv('VERSION')
+
+setup(name='ydata-synthetic',
+      version=VERSION,
+      description='Synthetic data generation methods with different synthetization methods.',
+      author='YData',
+      author_email='community@ydata.ai',
+      classifiers=[
+        'Programming Language :: Python :: 3 :: Only',
+        'Topic :: Software Development :: Artificial Intelligence :: Python Modules :: ',
+        'Operating System :: POSIX :: Linux',
+        'Operating System :: MacOS',
+        'Operating System :: Microsoft :: Windows',
+      ],
+      keywords='data science ydata',
+      url='https://github.com/ydataai/ydata-synthetic',
+      python_requires=">=3.6",
+      packages=find_namespace_packages('src'),
+      package_dir={'':'src'},
+      include_package_data=True,
+      install_requires=requirements,
+      options={"bdist_wheel": {"universal": True}})
diff --git a/models/cgan/__init__.py → src/__init__.py b/models/cgan/__init__.py → src/__init__.py
diff --git a/models/vanillagan/__init__.py → src/ydata_synthetic/__init__.py b/models/vanillagan/__init__.py → src/ydata_synthetic/__init__.py
diff --git a/models/wgan/__init__.py → ...ydata_synthetic/preprocessing/__init__.py b/models/wgan/__init__.py → ...ydata_synthetic/preprocessing/__init__.py
diff --git a/preprocessing/credit_fraud.py → ...a_synthetic/preprocessing/credit_fraud.py b/preprocessing/credit_fraud.py → ...a_synthetic/preprocessing/credit_fraud.py
diff --git a/src/ydata_synthetic/synthesizers/__init__.py b/src/ydata_synthetic/synthesizers/__init__.py
@@ -0,0 +1,9 @@
+from src.ydata_synthetic.synthesizers.regular.cgan.model import CGAN
+from src.ydata_synthetic.synthesizers.regular.wgan.model import WGAN
+from src.ydata_synthetic.synthesizers.regular.vanillagan.model import VanilllaGAN
+
+__all__ = [
+    "VanilllaGAN",
+    "CGAN",
+    "WGAN"
+]
diff --git a/models/gan.py → src/ydata_synthetic/synthesizers/gan.py b/models/gan.py → src/ydata_synthetic/synthesizers/gan.py
diff --git a/preprocessing/__init__.py → ...ynthetic/synthesizers/regular/__init__.py b/preprocessing/__init__.py → ...ynthetic/synthesizers/regular/__init__.py
diff --git a/src/ydata_synthetic/synthesizers/regular/cgan/__init__.py b/src/ydata_synthetic/synthesizers/regular/cgan/__init__.py
diff --git a/models/cgan/model.py → ...thetic/synthesizers/regular/cgan/model.py b/models/cgan/model.py → ...thetic/synthesizers/regular/cgan/model.py
@@ -160,7 +160,7 @@ def build_model(self, input_shape, dim):
 
 if __name__ == '__main__':
     import pandas as pd
-    from preprocessing.credit_fraud import transformations
+    from src.ydata_synthetic.preprocessing import transformations
     import sklearn.cluster as cluster
 
     data = pd.read_csv('/home/fabiana/PycharmProjects/YData/gan-playground/examples/data/creditcard.csv')

diff --git a/src/ydata_synthetic/synthesizers/regular/vanillagan/__init__.py b/src/ydata_synthetic/synthesizers/regular/vanillagan/__init__.py
diff --git a/models/vanillagan/model.py → .../synthesizers/regular/vanillagan/model.py b/models/vanillagan/model.py → .../synthesizers/regular/vanillagan/model.py
@@ -2,7 +2,7 @@
 from os import path
 import numpy as np
 
-from models import gan
+from src.ydata_synthetic.synthesizers import gan
 
 import tensorflow as tf
 from tensorflow.keras.layers import Input, Dense, Dropout

diff --git a/src/ydata_synthetic/synthesizers/regular/wgan/__init__.py b/src/ydata_synthetic/synthesizers/regular/wgan/__init__.py
diff --git a/models/wgan/model.py → ...thetic/synthesizers/regular/wgan/model.py b/models/wgan/model.py → ...thetic/synthesizers/regular/wgan/model.py
@@ -1,9 +1,8 @@
 import os
 from os import path
 import numpy as np
-from functools import partial
 
-from models import gan
+from src.ydata_synthetic.synthesizers import gan
 
 import tensorflow as tf
 from tensorflow.keras.layers import Input, Dense, Dropout