(feature_dims)
| 832 | |
| 833 | @staticmethod |
| 834 | def create_sparse_embeddings(feature_dims): |
| 835 | sparse_embeddings = [ |
| 836 | layers.SparseEmbedding(feature_dim + 1, 16) |
| 837 | for feature_dim in feature_dims |
| 838 | ] |
| 839 | return sparse_embeddings |
| 840 | |
| 841 | def __init__(self, metapath, fanouts, dim, |
| 842 | feature_ixs, feature_dims, shared_embeddings=None, |