scheduling_algorithm_didi/train_data/train_data.py

from entity import *
import random
import numpy as np


def get_data():
    all_data = []
    for id in range(20000):
        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.pkl")


if __name__ == '__main__':
    get_data()