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()