default_tester.py 文件源码

def RunTestor():
    # Create Volume test
    vXs, vYs = generateBaseVector(volume_filename, "volume")
    vXs_more = createVolumeVector(vXs, weather_filename)
    vXs_fin = generateProcessedVolumeVector(vXs_more)

    vX_train, vX_test, vy_train, vy_test = train_test_split(vXs_fin, vYs, test_size=0.1)
    volume_reg = DecisionTreeRegressor()
    volume_reg.fit(vX_train, vy_train)
    vResult = volume_reg.predict(vX_test)

    v_mape = VolumeMAPE(vX_test, vResult, vy_test)

    # Create Travel Time test
    tXs, tYs = generateBaseVector(travel_filename, "travel_time")
    tXs_more = createTravelTimeVector(tXs, weather_filename)
    tXs_fin = generateProcessedTravelTimeVector(tXs_more)

    tX_train, tX_test, ty_train, ty_test = train_test_split(tXs_fin, tYs, test_size=0.1)
    travelTime_reg = DecisionTreeRegressor()
    travelTime_reg.fit(tX_train, ty_train)
    tResult = travelTime_reg.predict(tX_test)

    t_mape = TravelTimeMAPE(tX_test, tResult, ty_test)

    print("MAPE of Volume Prediction: " + str(v_mape) + "\n")
    print("MAPE of Travel Prediction: " + str(t_mape) + "\n")
    return True