Files
2026-07-13 13:30:25 +08:00

13 lines
391 B
Matlab
Executable File

function [y, P] = logitMnPred(model, X)
% Prediction of multiclass (multinomial) logistic regression model
% Input:
% model: trained model structure
% X: d x n testing data
% Output:
% y: 1 x n predict label (1~k)
% P: k x n predict probability for each class
% Written by Mo Chen (sth4nth@gmail.com).
W = model.W;
X = [X; ones(1,size(X,2))];
P = softmax(W'*X);
[~, y] = max(P,[],1);