(self,
emb_stacking_columns=None,
dnn_hidden_units=[1024, 512, 256],
optimizer_type='adam',
cross_learning_rate=0.001,
deep_learning_rate=0.001,
inputs=None,
bf16=False,
stock_tf=None,
adaptive_emb=False,
input_layer_partitioner=None,
dense_layer_partitioner=None)
| 96 | |
| 97 | class DCN(): |
| 98 | def __init__(self, |
| 99 | emb_stacking_columns=None, |
| 100 | dnn_hidden_units=[1024, 512, 256], |
| 101 | optimizer_type='adam', |
| 102 | cross_learning_rate=0.001, |
| 103 | deep_learning_rate=0.001, |
| 104 | inputs=None, |
| 105 | bf16=False, |
| 106 | stock_tf=None, |
| 107 | adaptive_emb=False, |
| 108 | input_layer_partitioner=None, |
| 109 | dense_layer_partitioner=None): |
| 110 | if not inputs: |
| 111 | raise ValueError("Dataset is not defined.") |
| 112 | self._feature = inputs[0] |
| 113 | self._label = inputs[1] |
| 114 | |
| 115 | self._emb_stacking_columns = emb_stacking_columns |
| 116 | if not emb_stacking_columns: |
| 117 | raise ValueError("Embedding and stacking column is not defined.") |
| 118 | |
| 119 | self.tf = stock_tf |
| 120 | self.bf16 = False if self.tf else bf16 |
| 121 | self.is_training = True |
| 122 | self._adaptive_emb = adaptive_emb |
| 123 | |
| 124 | self._dnn_hidden_units = dnn_hidden_units |
| 125 | self._deep_learning_rate = deep_learning_rate |
| 126 | self._cross_learning_rate = cross_learning_rate |
| 127 | self._optimizer_type = optimizer_type |
| 128 | self._input_layer_partitioner = input_layer_partitioner |
| 129 | self._dense_layer_partitioner = dense_layer_partitioner |
| 130 | |
| 131 | self._create_model() |
| 132 | with tf.name_scope('head'): |
| 133 | self._create_loss() |
| 134 | self._create_optimizer() |
| 135 | self._create_metrics() |
| 136 | |
| 137 | # used to add summary in tensorboard |
| 138 | def _add_layer_summary(self, value, tag): |
nothing calls this directly
no test coverage detected