StanfordMLOctave/machine-learning-ex2/ex2/plotData.m

39 lines
1.0 KiB
Matlab

function plotData(X, y)
%PLOTDATA Plots the data points X and y into a new figure
% PLOTDATA(x,y) plots the data points with + for the positive examples
% and o for the negative examples. X is assumed to be a Mx2 matrix.
% Create New Figure
figure; hold on;
% ====================== YOUR CODE HERE ======================
% Instructions: Plot the positive and negative examples on a
% 2D plot, using the option 'k+' for the positive
% examples and 'ko' for the negative examples.
%
% Seperate between positive and negative ( 0 or 1 admission) to later represent on graph
pos = find(y == 1);
neg = find(y == 0);
% Setup Plots note: X(pos,1) finds all rows with positive output thanks to
% our find function returning a vector containing the indeces of each
plot(X(pos,1), X(pos,2), 'k+', 'LineWidth', 2, 'MarkerSize', 7);
plot(X(neg,1), X(neg,2), 'ko', 'MarkerFaceColor', 'y', 'MarkerSize', 7);
% =========================================================================
hold off;
end