It can be used to:
- scaffold new projects in seconds and customize only a minimum amount of code.
- import and optimize CSV datasets and train the model with them.
- visualize the model structure, loss and accuracy functions during training.
- export a simple REST API to use your models from a server.
Download the latest stable release, extract and:
cd /path/to/ergo sudo pip3 install -r requirements.txt python3 setup.py build sudo python3 setup.py install
Enable GPU support
Make sure you have CUDA 9.0 and cuDNN 7.0 installed and then:
sudo pip3 uninstall tensorflow sudo pip3 install tensorflow-gpu
Start by printing the available actions by running
ergo help, you can also print the software version (ergo, keras
and tensorflow versions) and some hardware info with
ergo info to verify your installation.
Once ready, create a new project named
ergo create example
Inside the newly created
example folder, there will be three files:
prepare.py, used to preprocess your dataset and inputs (if, for instance, you're using pictures instead of a csv file).
model.py, that you can change to customize the model.
train.py, for the training algorithm.
By default, ergo will simply read the dataset as a CSV file, build a small neural network with 10 inputs, two hidden layers of 30 neurons each and 2 outputs and use a pretty standard training algorithm. You can see a complete (and more complex) example on the planes-detector project repository.
After defining the model structure and the training process, you can import a CSV dataset (first column must be the label) and start training using 2 GPUs:
ergo train example --dataset /some/path/data.csv --gpus 2
This will split the dataset into a train, validation and test sets (partitioned with the
--validation arguments) and start the training.
If you want to update a model and/or train it on already imported data, you can simply:
ergo train example --gpus 2
Now it's time to visualize the model structure and how the the
loss metrics changed during training (requires
sudo apt-get install graphviz python3-tk):
ergo view example
Once you're done, you can remove the train, test and validation temporary datasets with:
ergo clean example
To load the model and start a REST API for evaluation (can be customized with
ergo serve example
You'll be able to access the model for evaluation via
To reset the state of a project (WARNING: this will remove the datasets, the model files and all training statistics):
ergo clean example --all
Evaluate and compare the performances of two trained models on a given dataset and (optionally) output the differences to a json file:
ergo cmp example_a example_b --dataset /path/to/data.csv --to-json diffs.json
Optimize a dataset (get unique rows and reuse 15% of the total samples, customize ratio with the
--reuse-ratio argument, customize output with
ergo optimize-dataset /some/path/data.csv
Convert the Keras model to frugally-deep format:
ergo to-fdeep example
ergo was made with ♥ by Simone Margaritelli and it is released under the GPL 3 license.