search

tensorlab / tensorfx

TensorFlow framework for training and serving machine learning models
Apache License 2.0
196 stars 41 forks source link

Implement Embedding feature #19

Open nikhilk opened 7 years ago

nikhilk commented 7 years ago

The input is one or more feature columns, and args are embedding output size, to produce the results of tf.nn.embedding_lookup().