def test_write_on_file(self): K = 10 # Output population pp = get_population() fout = open('pp.csv', 'w') fout.write('x,y,label\n') for x, y in pp: fout.write('%f,%f,k\n' % (x, y)) fout.close() # Output MDS samples lin = Linear(self.dist_func, K = K) lin.update(pp) sampled = lin.get_sampled() fout = open('mcs.csv', 'w') fout.write('x,y,label\n') for x, y in sampled: fout.write('%f,%f,b\n' % (x, y)) fout.close() # Output random samples random.seed(1) sampled = random.sample(pp, K) fout = open('random.csv', 'w') fout.write('x,y,label\n') for x, y in sampled: fout.write('%f,%f,g\n' % (x, y)) fout.close() # Ouptut k means samples kmedoids = KMedoids(self.dist_func, K) sampled = kmedoids.sample(pp) fout = open('kmedoids.csv', 'w') fout.write('x,y,label\n') for x, y in sampled: fout.write('%f,%f,r\n' % (x, y)) fout.close()
def _test_linear(self): K = 10 pp = get_population() lin = Linear(self.dist_func, K = K) for i in range(1): lin.update(pp) x, y = zip(*pp) plt.scatter(x, y, s=1, c='b') sampled = lin.get_sampled() x, y = zip(*sampled) plt.scatter(x, y, s=3000, c='r', alpha=0.2) #x, y = zip(*random.sample(pp, K)) #plt.scatter(x, y, s=3000, c='g', alpha=0.2) #kmedoids = KMedoids(self.dist_func, K) #x, y = zip(*kmedoids.sample(pp)) #plt.scatter(x, y, s=3000, c='b', alpha=0.2) plt.show()