(self, linear_feature_columns, dnn_feature_columns, cross_num=2, cross_parameterization='vector',
dnn_hidden_units=(128, 128), l2_reg_linear=0.00001, l2_reg_embedding=0.00001, l2_reg_cross=0.00001,
l2_reg_dnn=0, init_std=0.0001, seed=1024, dnn_dropout=0, dnn_activation='relu', dnn_use_bn=False,
task='binary', device='cpu', gpus=None)
| 42 | """ |
| 43 | |
| 44 | def __init__(self, linear_feature_columns, dnn_feature_columns, cross_num=2, cross_parameterization='vector', |
| 45 | dnn_hidden_units=(128, 128), l2_reg_linear=0.00001, l2_reg_embedding=0.00001, l2_reg_cross=0.00001, |
| 46 | l2_reg_dnn=0, init_std=0.0001, seed=1024, dnn_dropout=0, dnn_activation='relu', dnn_use_bn=False, |
| 47 | task='binary', device='cpu', gpus=None): |
| 48 | |
| 49 | super(DCN, self).__init__(linear_feature_columns=linear_feature_columns, |
| 50 | dnn_feature_columns=dnn_feature_columns, l2_reg_embedding=l2_reg_embedding, |
| 51 | init_std=init_std, seed=seed, task=task, device=device, gpus=gpus) |
| 52 | self.dnn_hidden_units = dnn_hidden_units |
| 53 | self.cross_num = cross_num |
| 54 | self.dnn = DNN(self.compute_input_dim(dnn_feature_columns), dnn_hidden_units, |
| 55 | activation=dnn_activation, use_bn=dnn_use_bn, l2_reg=l2_reg_dnn, dropout_rate=dnn_dropout, |
| 56 | init_std=init_std, device=device) |
| 57 | if len(self.dnn_hidden_units) > 0 and self.cross_num > 0: |
| 58 | dnn_linear_in_feature = self.compute_input_dim(dnn_feature_columns) + dnn_hidden_units[-1] |
| 59 | elif len(self.dnn_hidden_units) > 0: |
| 60 | dnn_linear_in_feature = dnn_hidden_units[-1] |
| 61 | elif self.cross_num > 0: |
| 62 | dnn_linear_in_feature = self.compute_input_dim(dnn_feature_columns) |
| 63 | |
| 64 | self.dnn_linear = nn.Linear(dnn_linear_in_feature, 1, bias=False).to( |
| 65 | device) |
| 66 | self.crossnet = CrossNet(in_features=self.compute_input_dim(dnn_feature_columns), |
| 67 | layer_num=cross_num, parameterization=cross_parameterization, device=device) |
| 68 | self.add_regularization_weight( |
| 69 | filter(lambda x: 'weight' in x[0] and 'bn' not in x[0], self.dnn.named_parameters()), l2=l2_reg_dnn) |
| 70 | self.add_regularization_weight(self.dnn_linear.weight, l2=l2_reg_linear) |
| 71 | self.add_regularization_weight(self.crossnet.kernels, l2=l2_reg_cross) |
| 72 | self.to(device) |
| 73 | |
| 74 | def forward(self, X): |
| 75 |
nothing calls this directly
no test coverage detected