【发布时间】:2021-03-08 07:08:19
【问题描述】:
我正在进行一个机器学习项目,我正在尝试预测英雄联盟游戏的获胜者。我现在正在训练 50,000 个游戏(我总共存储了 2000 万个数据点),我的模型得到的结果很差。
我的数据在一个 csv 文件中,前 5 列是代表一支球队的冠军 ID 的数字,接下来的 5 列是代表另一支球队冠军 ID 的数字,第 11 列是 1 或 0(一队胜1,一队负0)
它是使用 numpy 加载的:
dataset = loadtxt('/content/drive/MyDrive/Data/dataset.csv', delimiter=',')
有人可以帮忙吗?我对 Keras 不是很熟悉,所以我不确定我的模型本身是否需要修复,或者我的数据是否存在问题。
型号:
model.add(Dense(12, input_dim=10, activation='relu'))
model.add(Dense(8, activation='relu'))
model.add(Dense(1, activation='sigmoid'))
model.compile(loss='binary_crossentropy', optimizer='adam', metrics=['accuracy'])
这是我的准确率/损失图:
【问题讨论】:
-
能否请您描述/展示您正在使用的所有功能?
-
此外,对于 50K 数据点,该网络似乎太浅了。
-
@R.Schaefer 我只使用了上面提到的 csv 文件,我只是附上了一张截图,其中 1 行的样子,以及它是如何加载的。谢谢你的帮助。
-
@Frightera 您能否详细介绍解决此问题的最佳方法?我只是简单地添加更多层还是对如何最好地做到这一点有直觉。提前谢谢你
-
首先扩展您的数据。在尝试添加更多图层之后。您需要对这些层进行试验,没有直接的答案。
标签: python machine-learning keras data-science