一小时学会TensorFlow2之全连接层

下面是关于“一小时学会TensorFlow2之全连接层”的完整攻略。

全连接层简介

全连接层是神经网络中最基本的层之一，也是最常用的层之一。全连接层将输入数据与权重矩阵相乘，并加上偏置项，然后通过激活函数进行非线性变换，得到输出结果。

TensorFlow2中的全连接层

在TensorFlow2中，可以使用Dense层来创建全连接层。Dense层是一个可训练的层，可以自动学习权重矩阵和偏置项，并应用激活函数。

以下是使用Dense层创建全连接层的示例：

from tensorflow.keras.layers import Dense

model = Sequential()
model.add(Dense(64, activation='relu', input_dim=20))
model.add(Dense(10, activation='softmax'))

在上面的示例中，我们使用Keras创建了一个简单的神经网络模型，并使用Dense层创建了两个全连接层。第一个全连接层有64个神经元，使用ReLU激活函数，输入维度为20。第二个全连接层有10个神经元，使用softmax激活函数。

示例1：使用全连接层进行分类

以下是使用全连接层进行分类的示例：

from tensorflow.keras.layers import Dense
from tensorflow.keras.models import Sequential
from tensorflow.keras.utils import to_categorical
import numpy as np

# Generate dummy data
x_train = np.random.random((1000, 20))
y_train = to_categorical(np.random.randint(10, size=(1000, 1)), num_classes=10)

x_test = np.random.random((100, 20))
y_test = to_categorical(np.random.randint(10, size=(100, 1)), num_classes=10)

# Define model
model = Sequential()
model.add(Dense(64, activation='relu', input_dim=20))
model.add(Dense(10, activation='softmax'))

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

# Train model
model.fit(x_train, y_train, epochs=10, batch_size=32, validation_data=(x_test, y_test))

在上面的示例中，我们使用全连接层创建了一个简单的分类模型，并使用categorical_crossentropy作为损失函数，使用adam作为优化器，使用accuracy作为评估指标。

示例2：使用全连接层进行回归

以下是使用全连接层进行回归的示例：

from tensorflow.keras.layers import Dense
from tensorflow.keras.models import Sequential
import numpy as np

# Generate dummy data
x_train = np.random.random((1000, 20))
y_train = np.random.random((1000, 1))

x_test = np.random.random((100, 20))
y_test = np.random.random((100, 1))

# Define model
model = Sequential()
model.add(Dense(64, activation='relu', input_dim=20))
model.add(Dense(1, activation='linear'))

# Compile model
model.compile(loss='mean_squared_error', optimizer='adam')

# Train model
model.fit(x_train, y_train, epochs=10, batch_size=32, validation_data=(x_test, y_test))

在上面的示例中，我们使用全连接层创建了一个简单的回归模型，并使用mean_squared_error作为损失函数，使用adam作为优化器。