获取神经网络的预测列表

时间:2019-10-28 17:15:37

标签: python neural-network

我创建了一个神经网络来对消息进行分类。现在,我想将预测收集到python中的列表中。我该怎么做?

这是模型:

  var lines = File.ReadAllLines(_file);
  var firstOccurrence = lines.FirstOrDefault(l => l.Contains(contains));

在这里,我尝试打印在model.fit()方法内部的validation_data的形状,但它给出了一个错误。

model = Sequential()
model.add(layers.Dense(500, activation = "relu", input_shape=(7600,)))
# Hidden - Layers
model.add(layers.Dropout(0.4, noise_shape=None, seed=None))
model.add(layers.Dense(300, activation = "relu"))
model.add(layers.Dropout(0.4, noise_shape=None, seed=None))
model.add(layers.Dense(100, activation = "relu"))
model.add(layers.Dropout(0.4, noise_shape=None, seed=None))

model.add(layers.Dense(20, activation = "softmax"))
model.summary()

model.compile(loss="categorical_crossentropy",
              optimizer="adam",
              metrics=['accuracy'])



model.fit( np.array(vectorized_training), np.array(y_train_neralnet),
          batch_size=2000,
          epochs=3,
          verbose=1,
          validation_data=(np.array(vectorized_validation), np.array(y_validation_neralnet)))

1 个答案:

答案 0 :(得分:1)

这是您要寻找的:

3