# 导入必要的库
import numpy as np
import os
from PIL import Image
from sklearn.model_selection import KFold
from sklearn.preprocessing import StandardScaler
from sklearn.neural_network import MLPClassifier
from sklearn.metrics import accuracy_score
# 加载图像数据
def load_image_data(image_dir, target_size=(64, 64)):
X = []
y = []
class_labels = os.listdir(image_dir)
for label in class_labels:
class_dir = os.path.join(image_dir, label)
if os.path.isdir(class_dir):
for file_name in os.listdir(class_dir):
file_path = os.path.join(class_dir, file_name)
if file_path.endswith(('.png', '.jpg', '.jpeg')):
image = Image.open(file_path)
image = image.resize(target_size)
image = np.array(image)
X.append(image)
y.append(int(label))
return np.array(X), np.array(y)
# 加载数据
image_dir = 'path/to/your/image/directory' # 替换为你的图像数据路径
X, y = load_image_data(image_dir)
# 数据预处理:将图像数据展开为一维向量,并标准化数据
X = X.reshape((X.shape[0], -1))
scaler = StandardScaler()
X = scaler.fit_transform(X)
# 定义多层感知器模型
model = MLPClassifier(hidden_layer_sizes=(100,), max_iter=300, random_state=42)
# 定义五折交叉验证
kf = KFold(n_splits=5, shuffle=True, random_state=42)
accuracy_list = []
for train_index, test_index in kf.split(X):
X_train, X_test = X[train_index], X[test_index]
y_train, y_test = y[train_index], y[test_index]
# 训练多层感知器模型
model.fit(X_train, y_train)
# 评估模型
y_pred = model.predict(X_test)
accuracy = accuracy_score(y_test, y_pred)
accuracy_list.append(accuracy)
# 输出平均准确率
print(f'平均准确率: {np.mean(accuracy_list):.4f}')
多层感知机MLP
于 2024-07-02 19:53:46 首次发布