% Load variables t_data and X_data
load('data.mat');

% Initialize a gridfor k
G = 60;
h = linspace(1,6,G);
[K1, K2] = meshgrid(h);
SSE = zeros(size(K1));

% Loop through the grid points
for i = 1:G
    for  j = 1:G
        k_ij = [K1(i,j), K2(i,j)];
        SSE(i,j) = compute_SSE(k_ij, t_data, X_data);
    end
end

% Surface plot of log(SSE)
subplot(1,2,1);
surf(K1, K2, log(SSE), 'EdgeColor', 'none');
colorbar;
xlabel('k_1');
ylabel('k_2');
zlabel('log(SSE)');

% Contour plot of log(SSE)
subplot(1,2,2);
contour(K1, K2, log(SSE), 30);
colorbar;
xlabel('k_1');
ylabel('k_2');