Skip to content
Snippets Groups Projects
Commit 2c26f7da authored by Tiago de Freitas Pereira's avatar Tiago de Freitas Pereira
Browse files

Introduced the Embeddings

parent 5ddba4e0
Branches
Tags
No related merge requests found
Pipeline #
#!/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]
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment