人工智能与大数据技术导论
下面来创建一个神经网络,通过DNNRegressor类来完成。我们使用hidden units来定义神经网络的结构。hiddenunits提供了一个整数列表,每个整数对应一个隐藏层,代表隐藏层中节点的个数,比如:
上面指定了包含2个隐藏层的神经网络。第一个隐藏层包含3个节点,第二个隐藏层包含10个节点。如果我们想添加更多的层,就可以添加更多的整数到列表中。比如,hidden\_units=[10,20,30,40]将创建4层,分别包含10、20、30和40个节点。默认情况下,所有隐藏层使用ReLU激活函数,并完全连接。
定义特征列和输入函数:
314