datasort.m

来自「这是一个支持向量机的工具」· M 代码 · 共 26 行

function [XYsort,Xlabel,Ylabel,Xsort,Ysort,U]=datasort(Xtrain,Ytest,X_hat,Y_hat,Z)
% sorting order of data according to values of variable Z

[S,nz]=size(Xtrain);
[R,nz]=size(Ytest);
m=[1:S]';
n=[S+(1:R)]';
U=[Xtrain X_hat];         % extended training set with prediction values
Q=[Ytest Y_hat];          % extended test set with prediction values
U=[U;Q];                  % data set with whole data

if Z>0
[Q,R]=sort(U(:,Z));       % sorting as per values of Zth variable
XYsort=U(R,:);
Xlabel=find(R<S+1);       % data labels of training set
Ylabel=find(R>S);         % data labels of test set
Xsort=XYsort(Xlabel,1:nz);% ordered training set
Ysort=XYsort(Ylabel,1:nz);% ordered test set
U=R(Ylabel)-S;
else
XYsort=U;
Xlabel=m;
Ylabel=n;
Xsort=Xtrain;
Ysort=Ytest;
end