Estoy utilizando la función plotgauss2d de BNT para visualizar cómo cambia la respuesta de un nodo Gaussiano 2D cuando se observa evidencia en otros lugares de la red.

eng = jtree_inf_engine(bnet);
evidence = cell(1, 2)
eng = enter_evidence(eng, evidence);
marginals = marginal_nodes(eng, 1); p_1 = marginals.T
marginals = marginal_nodes(eng, 2); p_2 = marginals.T
marginals
clf; plotgauss2d(marginals.mu, marginals.Sigma);
hold all;
evidence{1} = 1;
marginals = marginal_nodes(enter_evidence(eng, evidence), 2);
p = plotgauss2d(marginals.mu, marginals.Sigma);
set(p, 'Color', 'green');
evidence{1} = 2;
marginals = marginal_nodes(enter_evidence(eng, evidence), 2);
p = plotgauss2d(marginals.mu, marginals.Sigma);
set(p, 'Color', 'red');
legend({'Unknown', 'Class 1', 'Class 2'});
hold off;

Como puede ver, la leyenda no recoge los colores de trazado modificados que tuve que configurar manualmente. (Lamentablemente, plotgauss2d no alterna los colores automáticamente, como desearía).

¿Hay una manera de establecer los colores de línea utilizados en la leyenda también?