Predicting Fine-Grained Traffic Conditions via Spatio-Temporal LSTM

Algorithm 3

To generated the segment-based traffic data.

Input: Road Log , GPS Log ,

Output: The traffic data about the road segments.

(1) Initialize , , , , ,

, , ,;

(2) GetVehicleColumns(ID, Time, V_Speed, V_lon, V_lat);

(3) GetRoadColumns(RoadID, R_lon, R_lat);

(4) Repeat ;

(5) ;

(6) For to

(7) if

(8) , ;

(9) , ;

(10) , ;

(11) , ;

(12) then

(13)

(14) ;

(15)

(16) ;

(17)

(18) Imitate GetCountData(, );

(19) Imitate GetTTData(, );

(20) if then ;

(21) End for;

(22) ;

(23) Imitate GetData(, ,

);

(24) ;

(25) ;

(26) return, , ,

, , .