Skip to content

Instantly share code, notes, and snippets.

View jostmey's full-sized avatar

Jared Ostmeyer jostmey

View GitHub Profile
@jostmey
jostmey / NDT.py
Created September 18, 2022 15:43
##########################################################################################
# Author: Jared L. Ostmeyer
# Date Started: 2020-08-13
# Purpose: Implementation neural decision tree (NDT) and variants for Keras in TensorFlow
##########################################################################################
from tensorflow.keras.layers import *
from tensorflow.keras.initializers import Initializer
import tensorflow as tf
@jostmey
jostmey / RWACell.py
Last active September 8, 2018 21:05
Recurrent Weighted Average (RWA) model as a Tensorflow RNNCell
##########################################################################################
# Author: Jared L. Ostmeyer
# Date Started: 2017-04-11
# Purpose: Recurrent weighted average cell for tensorflow.
##########################################################################################
"""Implementation of recurrent weighted average cell as a TensorFlow module. See
https://arxiv.org/abs/1703.01253 for a mathematical description of the model.
"""