import numpy as np
import tensorflow as tf
from tensorflow.contrib import layers
from config import MovieQAPath
from legacy.input_v2 import Input
_mp = MovieQAPath()
hp = {'emb_dim': 256, 'feat_dim': 512, 'dropout_rate': 0.1}
reg = layers.l2_regularizer(0.3)
def dropout(x, training):
return tf.layers.dropout(x, hp['dropout_rate'], training=training)
def make_mask(x, length):
return tf.tile(tf.expand_dims(tf.sequence_mask(x, maxlen=length), axis=-1),
[1, 1, hp['emb_dim']])
def mask_tensor(x, mask):
zeros = tf.zeros_like(x)
x = tf.where(mask, x, zeros)
return x
def l2_norm(x, axis=1):
return tf.nn.l2_normalize(x, axis=axis)
def setUp(self):
self.model = BasicModel()
self.model._hp.parse()
self.mp = MovieQAPath()
def __init__(self):
self._hp = BasicHP()
self._mp = MovieQAPath()
