Method __init__

model_demo/DeepCrossing/layer.py:39–43

(self, k, sparse_feature_columns)

class Embed_layer(Layer):
    def __init__(self, k, sparse_feature_columns):
        super(Embed_layer, self).__init__()
        self.emb_layers = [
            Embedding(feat["feat_onehot_dim"], k) for feat in sparse_feature_columns
        ]

    def call(self, inputs, **kwargs):
        if K.ndim(inputs) != 2: