setup.py

from distutils.core import setup

setup(
    name='ML_toolkit',
    version='1.0',
    packages=[
        'dataset_utils',
        'sequence_model',
        'computer_vision',
        'variable_selection',
        'gcp_example',
        'gcp_image_classification'
    ],
    package_dir={
        'sequence_model': 'src/model/sequence_model',
        'computer_vision': 'src/model/computer_vision',
        'dataset_utils': 'src/processing/dataset_utils',
        'variable_selection': 'src/processing/variable_selection',
        'gcp_example': 'src/cloud_tools/gcp/example',
        'gcp_image_classification': 'src/cloud_tools/gcp/image_classification'
    },
    author='Dorian Bagur',
    author_email='dorian.bagur@gmail.com',
    description='Machine learning models and processing'
)