search

leopiney / deep-forest

An on going implementation of the gcForest algorithm
MIT License
194 stars 58 forks source link

Improvements for training CIFAR10 #2

Closed leopiney closed 7 years ago

leopiney commented 7 years ago
  1. Several memory usage improvements Use generators to create windows instead of creating all of them at once.
  2. Add helpers to work with CIFAR10 Run sh get_datasets.sh to download the CIFAR10 data, and use the load_CIFAR10 method from the data_utils.py module to get the training and test data from the CIFAR10 dataset.