from sklearn.neighbors import NearestNeighbors
#-----------------------------------
if __name__ == "__main__":
random.seed( 12345 )
# como_extract.computeHistogramsAllBuses(); exit(0)
(all_buses, periods_all_buses) = Util.pickleLoad(DATA_FILE_NAME+"_"+SIGNAL_CODE+".txt")
dates_all_buses = [ [ Util.getDate(tm) for tm in times ] for times in periods_all_buses ]
# como_ploting.Ploting.vizualize_buses(all_buses, dates_all_buses, m = '.'); exit(0)
#-----------------------------------
for id_bus in range( len(all_buses) ): # for each test bus
h = IGNG( radius = 2*PARAMS["R"] ); h.train( [Util.centroid( Util.flatList(all_buses) )] ); dir_imgs = "cosmo_IGNG_deriv_mature/"
# h = GNG(period = 200); h.train( [Util.centroid( Util.flatList(all_buses) )] ); dir_imgs = "cosmo_GNG/"
own_test = all_buses[id_bus]
fleet_test = all_buses[:id_bus] + all_buses[id_bus+1 :]
filename = DBFILES[id_bus]
busname = filename.split("_")[0]
dates = dates_all_buses[id_bus]
Z1 = []; Z2 = []; S1 = []; S2 = []
#--------------------------
for i, his_test in enumerate( own_test ): # for each day
sys.stdout.write( "\r%s" % "---------------------------- progress = " + str(i*100./len(own_test)) + " " + DBFILES[id_bus] + " " ); sys.stdout.flush()
