text_classification_model_simple.py 文件源码

def model_fully_connected(self, output, gene, variation, num_output_classes, dropout, training):
        output = layers.dropout(output, keep_prob=dropout, is_training=training)
        net = tf.concat([output, gene, variation], axis=1)
        net = layers.fully_connected(net, 128, activation_fn=tf.nn.relu)
        net = layers.dropout(net, keep_prob=dropout, is_training=training)
        logits = layers.fully_connected(net, num_output_classes, activation_fn=None)
        return logits