我正在尝试实现 NN ML 算法来将产品映射到各自的类别，但该算法没有给出一致的结果。我使用了 keras、tensorflow 的顺序模型。请提出建议...

我正在尝试实现 NN ML 算法来将产品映射到各自的类别，但该算法没有给出一致的结果。我使用了 keras、tensorflow 的顺序模型。请提出是否有更好的方法来解决这个问题。

代码片段：

 # Split into training and test sets
     X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)
     y_train = to_categorical(y_train)
     y_test = to_categorical(y_test)
     print(y_train)
 # Define the model
     model = Sequential()
     model.add(Embedding(input_dim=5000, output_dim=128, input_length=max_len))
     model.add(LSTM(units=128, dropout=0.2, recurrent_dropout=0.2))
     model.add(Dense(units=128, activation='relu'))
     model.add(Dropout(0.5))
     model.add(Dense(units=y_train.shape[1], activation='softmax'))
 
     # Compile the model
     model.compile(optimizer='adam', loss='categorical_crossentropy', metrics=['accuracy']

)