Update _Dist/NeuralNetworks

Author: carefree0910

_Dist/NeuralNetworks/e_AdvancedNN/NN.py

@@ -97,7 +97,7 @@ def init_model_structure_settings(self):
     def _get_embedding(self, i, n):
         embedding_size = math.ceil(math.log2(n)) + 1 if self.embedding_size == "log" else self.embedding_size
         embedding = tf.Variable(tf.truncated_normal(
-            [1, embedding_size], mean=0, stddev=0.02
+            [n, embedding_size], mean=0, stddev=0.02
         ), name="Embedding{}".format(i))
         return tf.nn.embedding_lookup(embedding, self._categorical_xs[i], name="Embedded_X{}".format(i))