神经网络

来源:6-12 回归-分类-人工神经网络-2

哲川恒五郎

2018-09-13

老师,
人工神经网络那一讲中,我有些不懂得地方。
mdl.add(Dense(50, input_dim=len(f_v[0])))
input_dim是输入维度,这里选的是第一个变量,输入维度具体是什么意思?

mdl.fit(X_train, np.array([[0,1] if i==1 else [1,0] for i in Y_train]),nb_epoch=10000, batch_size=8999)
关于np.array那一块能具体解释一下吗?

写回答

2回答

途索

2018-09-13

[[0,1] if i==1 else [1,0] for i in Y_train],这块的意思是,如果Y_train中是1的话,就替换成[0,1];否则(即此例中的Y_train中的元素是0),就替换成[1,0]。这么变换的作用,是因为在神经网络中,输出的维度必须和要分类的种类数相等。此例中,要判断的是两类,所以输出一定是要表示成One-Hot形式。同理,如果要输出三类,那一定要转换成[1,0,0],[0,1,0],[0,0,1]这样的形式。

2
0

途索

2018-09-13

同学你好,输入维度填的是输入的特征数,比如一条数据有10个特征,这里就填10,因为例子里f_v队列里,每一个元素都是一条数据,所以len(f_v[0])就可以获得数据的特征数量了,f_v的长度是数据的数量,f_v中每一条数据的长度就是特征的数量了。

0
0

Python3数据分析与挖掘建模实战,快速胜任数据分析师

快速胜任数据分析岗位,逆袭成为数据掘金时代的抢手人才!

2204 学习 · 489 问题

查看课程