Commit 2c26f7da authored by Tiago de Freitas Pereira's avatar Tiago de Freitas Pereira
Browse files

Introduced the Embeddings

parent 5ddba4e0
Pipeline #8138 failed with stages
in 9 minutes and 51 seconds
#!/usr/bin/env python
# vim: set fileencoding=utf-8 :
# @author: Tiago de Freitas Pereira <tiago.pereira@idiap.ch>
# @date: Wed 11 May 2016 09:39:36 CEST
import tensorflow as tf
from bob.learn.tensorflow.utils.session import Session
class Embedding(object):
"""
Embedding
"""
def __init__(self, input, graph):
self.input = input
self.graph = graph
def __call__(self, data):
session = Session.instance().session
feed_dict = {self.input: data}
return session.run([self.graph], feed_dict=feed_dict)[0]
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment