Imported from https://github.com/wbengine/TRF-NN-Tensorflow