74 lines
2.2 KiB
Mathematica
74 lines
2.2 KiB
Mathematica
|
exercice_3;
|
||
|
|
|
||
|
|
% Valeurs initiales des proportions :
|
||
|
|
proportion_1 = size(D_app_1,2)/size(D_app,2);
|
||
|
|
proportion_2 = 1-proportion_1;
|
||
|
|
|
||
|
|
% Algorithme EM :
|
||
|
|
difference_score = 1;
|
||
|
|
seuil = 0.000001;
|
||
|
|
i = 1;
|
||
|
|
pause(2)
|
||
|
|
while abs(difference_score)>seuil
|
||
|
|
|
||
|
|
% Calcul des probabilités d'appartenance aux deux classes :
|
||
|
|
parametres_estim = [parametres_1_estim ; parametres_2_estim];
|
||
|
|
probas = probabilites_EM(D_app,parametres_estim,proportion_1,proportion_2,sigma);
|
||
|
|
probas_1 = probas(1,:);
|
||
|
|
probas_2 = probas(2,:);
|
||
|
|
|
||
|
|
% Partition des données :
|
||
|
|
classe_1 = find(probas_1>=probas_2);
|
||
|
|
classe_2 = find(probas_1<probas_2);
|
||
|
|
D_app_1 = D_app(:,classe_1);
|
||
|
|
D_app_2 = D_app(:,classe_2);
|
||
|
|
|
||
|
|
% Affichage de la partition :
|
||
|
|
hold off;
|
||
|
|
plot(D_app_1(1,:),D_app_1(2,:),'+b','MarkerSize',10,'LineWidth',2);
|
||
|
|
hold on;
|
||
|
|
plot(D_app_2(1,:),D_app_2(2,:),'+g','MarkerSize',10,'LineWidth',2);
|
||
|
|
|
||
|
|
% Mise à jour des proportions :
|
||
|
|
proportion_1 = mean(probas_1);
|
||
|
|
proportion_2 = 1-proportion_1;
|
||
|
|
|
||
|
|
% Estimation en moindres carrés pondérés :
|
||
|
|
X_1 = moindres_carres_ponderes(D_app,probas_1);
|
||
|
|
parametres_1_estim = conversion(X_1);
|
||
|
|
X_2 = moindres_carres_ponderes(D_app,probas_2);
|
||
|
|
parametres_2_estim = conversion(X_2);
|
||
|
|
|
||
|
|
% Tracé des ellipses estimées en moindres carrés (traits bleu et vert) :
|
||
|
|
[x_1,y_1] = points_ellipse(parametres_1_estim,theta_affichage);
|
||
|
|
plot([x_1 x_1(1)],[y_1 y_1(1)],'b-','LineWidth',3);
|
||
|
|
[x_2,y_2] = points_ellipse(parametres_2_estim,theta_affichage);
|
||
|
|
plot([x_2 x_2(1)],[y_2 y_2(1)],'g-','LineWidth',3);
|
||
|
|
legend(' Classe 1',' Classe 2',' Ellipse 1',' Ellipse 2','Location','Best');
|
||
|
|
|
||
|
|
xlabel('$x$','Interpreter','Latex');
|
||
|
|
ylabel('$y$','Interpreter','Latex');
|
||
|
|
axis([-taille taille -taille taille]);
|
||
|
|
axis equal;
|
||
|
|
|
||
|
|
% Calcul du nouveau score :
|
||
|
|
score_nouv = calcul_score_2(parametres_1_VT,parametres_2_VT,parametres_1_estim,parametres_2_estim);
|
||
|
|
difference_score = score_nouv-score;
|
||
|
|
score = score_nouv;
|
||
|
|
pause(1);
|
||
|
|
|
||
|
|
F = getframe(gcf);
|
||
|
|
[X, Map] = frame2im(F);
|
||
|
|
[fimage, map] = rgb2ind(X, 256);
|
||
|
|
if i == 1
|
||
|
|
imwrite(fimage, map, "saves/exo4.gif", 'gif', 'Loopcount', inf);
|
||
|
|
else
|
||
|
|
imwrite(fimage, map, "saves/exo4.gif", 'gif', 'WriteMode', 'append');
|
||
|
|
end
|
||
|
|
|
||
|
|
i = i + 1;
|
||
|
|
end
|
||
|
|
|
||
|
|
% Affichage du score final :
|
||
|
|
fprintf('Score de l''estimation par EM : %.3f\n',score);
|