cmu-db / peloton

The Self-Driving Database Management System
http://pelotondb.io
Apache License 2.0
2.03k stars 622 forks source link

Add AugmentedNN in optimizer to predict selectivity for range predicates #1472

Closed yetiancn closed 6 years ago

yetiancn commented 6 years ago

I add python scripts and cpp wrapper for the model, taking lstm.py and lstm_tf.cpp as a reference. Test cases for the model are also added. The test cases generate a uniform distribution dataset and a skewed distribution dataset. Hyperparameters, especially number of training epochs, need to be discussed based on real system experiments.