Method _build_ctr_model

modelzoo/esmm/train.py:303–316

(self, net)

        return net

    def _build_ctr_model(self, net):
        with tf.variable_scope('ctr_mlp',
                               partitioner=self._dense_layer_partitioner,
                               reuse=tf.AUTO_REUSE):
            for layer_id, num_hidden_units in enumerate(self._ctr_mlp):
                net = self._create_dense_layer(net,
                                                num_hidden_units,
                                                tf.nn.relu,
                                                f'ctr_mlp_hiddenlayer_{layer_id}')
            net = self._create_dense_layer(net,
                                            1,
                                            tf.math.sigmoid,
                                            'ctr_mlp_hiddenlayer_last')
        return net

# generate dataset pipline
def build_model_input(filename, batch_size, num_epochs, seed, stock_tf, workqueue=None):