def testTwoClsDiscriminator(): gmm = MCL.composeGMM(3) (t, x) = gmm.sample().mixtures() colors = [['blue', 'red'][int(label)] for label in t] plt.scatter(x[:, 0], x[:, 1], color=colors) disc = twoClsDiscriminator() disc.fit(x, t) xlim = [min(x[:, 0]), max(x[:, 0])] ylim = [min(x[:, 1]), max(x[:, 1])] tics = arange(xlim[0] - 3, xlim[1] + 3, 0.01) plt.xlim(xlim[0] - 3, xlim[1] + 3) plt.ylim(ylim[0] - 3, ylim[1] + 3) plt.scatter(*disc.clusterCenters(), color='black') plt.plot(*disc.centerDiffPlane(tics), color='gray') plt.plot(*disc.wPlane(tics), color='orange') plt.show()
def testTwoClsDiscriminator(): gmm = MCL.composeGMM(3) (t,x) = gmm.sample().mixtures() colors = [['blue', 'red'][int(label)] for label in t] plt.scatter(x[:,0], x[:,1], color=colors) disc = twoClsDiscriminator(); disc.fit(x,t) xlim = [min(x[:,0]), max(x[:,0])] ylim = [min(x[:,1]), max(x[:,1])] tics = arange(xlim[0]-3,xlim[1]+3,0.01) plt.xlim(xlim[0]-3,xlim[1]+3) plt.ylim(ylim[0]-3,ylim[1]+3) plt.scatter(*disc.clusterCenters(), color='black') plt.plot(*disc.centerDiffPlane(tics), color='gray') plt.plot(*disc.wPlane(tics), color='orange') plt.show()
def main(): gmm = MCL.composeGMM(type)