from entity import *
import random
import numpy as np


def get_data():
    all_data = []
    for id in range(200000):
        day_data = []
        for i in range(random.randint(5,25)):
            driver = Driver(id,random.randint(0,49),random.randint(0,49))
            to_x = random.randint(0,49)
            to_y = random.randint(0,49)

            order = Order(0,random.randint(0,49),random.randint(0,49),to_x,to_y,random.randint(1,144))
            match = Match(order,driver)
            day_data.append(match)

        all_data.append(day_data)
    save(all_data,"train_data_02.pkl")


if __name__ == '__main__':
    get_data()
 #    q = [[6.3724101e-02 ,1.0513991e+02],
 # [1.9977689e-03 ,9.5939758e+01],
 # [1.5849888e-02, 1.0017528e+02]]
 #    print(np.max(q, axis=1))