Skip to main content

Automation of the creation of the architecture of the neural network based on the input

Project description

Auto-Deep-Learning (Auto Deep Learning)

Downloads Version Python-Version issues PyPI - Status License

auto_deep_learning: with this package, you will be able to create, train and deploy neural networks automatically based on the input that you provide.

Installation

Use the package manager pip to install auto_deep_learning.

To install the package:

    pip install auto_deep_learning

If using an old version of the package, update it:

    pip install --upgrade auto_deep_learning

Basic Usage

Dataset

The data that it expects is a pd.DataFrame(), where the columns are the following:

    - image_path: the path to the image
    - class1: the classification of the class nr. 1. For example: {t-shirt, glasses, ...}
    - class2: the classification of the class nr. 2. For example: {summer, winter, ...}
    - ...
    - split_type: whether it is for training/validation/testing

For better performance, it is suggested that the classes and the type are of dtype category in the pandas DataFrame. If the type is not provided in the dataframe, you should use the utils function of data_split_types (in utils.dataset.sampler file).

If instead you have the images ordered in the structure of ImageFolder, which is the following structure:

    train/  
        class1_value/
            1.jgp
            2.jpg
            ...
        class2_value/
            3.jpg
            4.jpg
            ...
    test/
        class1_value/
            1.jgp
            2.jpg
            ...
        class2_value/
            3.jpg
            4.jpg
            ...

For simplifying logic, we have provided a logic that gives you the expected dataframe that we wanted, with the function of image_folder_convertion (in utils.functions), where it is expecting a path to the parent folder where the train/ and /test folders are.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

auto_deep_learning-0.1.4.8.tar.gz (8.4 kB view hashes)

Uploaded Source

Built Distribution

auto_deep_learning-0.1.4.8-py3-none-any.whl (16.3 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page