data_process/split.py

import pandas as pd

allowed = ["CA 134", "CA 170","I 605"]

df = pd.read_csv("/Users/amitomer/Desktop/Personal/University/deep_learning/TrafCast/old_data/split_df_v2.csv")

df = df[df["road_name"].isin(allowed)]

df.to_csv("weights_test.csv", index=False)