macbert/test.py

import pandas as pd
import numpy as np

ann_file = '/home/jchsiao/Desktop/macbert_code/dataset/travel_train_8000.csv'
data = np.array(pd.read_csv(ann_file))
print(data[1, 1])