鸢尾花分类matlab程序
时间: 2023-11-02 22:52:49 浏览: 206
MATLAB: 神经网络预测鸢尾花的分类
5星 · 资源好评率100%
以下是一个简单的鸢尾花分类matlab程序,使用K近邻算法实现:
%% 加载数据集
load fisheriris.mat
X = meas;
Y = species;
%% 划分训练集和测试集
cv = cvpartition(Y,'Holdout',0.3);
Xtrain = X(training(cv),:);
Ytrain = Y(training(cv));
Xtest = X(test(cv),:);
Ytest = Y(test(cv));
%% 训练K近邻模型
Mdl = fitcknn(Xtrain,Ytrain,'NumNeighbors',5);
%% 预测测试集
Ypred = predict(Mdl,Xtest);
%% 统计预测结果
accuracy = mean(Ypred == Ytest);
disp(['Accuracy: ' num2str(accuracy)]);
%% 可视化结果
gscatter(Xtest(:,1),Xtest(:,2),Ypred);
hold on;
gscatter(Xtest(:,1),Xtest(:,2),Ytest,'k','ox');
legend('Predicted','Actual');
xlabel('Sepal length');
ylabel('Sepal width');
title('Iris Classification');
阅读全文