diff options
author | Étienne Simon <esimon@esimon.eu> | 2015-07-23 17:48:38 -0400 |
---|---|---|
committer | Étienne Simon <esimon@esimon.eu> | 2015-07-23 17:48:38 -0400 |
commit | 1ee965e32c6678ca063d853d80d7bdeae0ab9734 (patch) | |
tree | 859085f379c3a24590c234be1fab3fb03272619a /data/cuts | |
parent | 66f2fc543fda43ad3fb05750e12121991d652685 (diff) | |
download | taxi-1ee965e32c6678ca063d853d80d7bdeae0ab9734.tar.gz taxi-1ee965e32c6678ca063d853d80d7bdeae0ab9734.zip |
Add Train/Valid/Test set
Diffstat (limited to 'data/cuts')
-rw-r--r-- | data/cuts/tvt_test.py | 9 | ||||
-rw-r--r-- | data/cuts/tvt_valid.py | 9 |
2 files changed, 18 insertions, 0 deletions
diff --git a/data/cuts/tvt_test.py b/data/cuts/tvt_test.py new file mode 100644 index 0000000..cd97a2b --- /dev/null +++ b/data/cuts/tvt_test.py @@ -0,0 +1,9 @@ +import random + +begin = 1372636853 +end = 1404172787 + +random.seed(42) +cuts = [] +for i in range(500): + cuts.append(random.randrange(begin, end)) diff --git a/data/cuts/tvt_valid.py b/data/cuts/tvt_valid.py new file mode 100644 index 0000000..c5bb828 --- /dev/null +++ b/data/cuts/tvt_valid.py @@ -0,0 +1,9 @@ +import random + +begin = 1372636853 +end = 1404172787 + +random.seed(1337) +cuts = [] +for i in range(500): + cuts.append(random.randrange(begin, end)) |