14 lines
433 B
Python
14 lines
433 B
Python
|
import pandas as pd
|
||
|
|
from sklearn.model_selection import train_test_split
|
||
|
|
|
||
|
|
TEST_SIZE = 25
|
||
|
|
VALID_SIZE = 25
|
||
|
|
|
||
|
|
data = pd.read_csv('./stop_times.normalized.tsv', sep='\t')
|
||
|
|
train, test = train_test_split(data, test_size=TEST_SIZE+VALID_SIZE)
|
||
|
|
valid, test = train_test_split(test, test_size=TEST_SIZE)
|
||
|
|
|
||
|
|
train.to_csv('stop_times.train.tsv', sep='\t')
|
||
|
|
test.to_csv('stop_times.test.tsv', sep='\t')
|
||
|
|
valid.to_csv('stop_times.valid.tsv', sep='\t')
|