aboutsummaryrefslogtreecommitdiff
path: root/data/cuts/tvt_test.py
diff options
context:
space:
mode:
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
commit1ee965e32c6678ca063d853d80d7bdeae0ab9734 (patch)
tree859085f379c3a24590c234be1fab3fb03272619a /data/cuts/tvt_test.py
parent66f2fc543fda43ad3fb05750e12121991d652685 (diff)
downloadtaxi-1ee965e32c6678ca063d853d80d7bdeae0ab9734.tar.gz
taxi-1ee965e32c6678ca063d853d80d7bdeae0ab9734.zip
Add Train/Valid/Test set
Diffstat (limited to 'data/cuts/tvt_test.py')
-rw-r--r--data/cuts/tvt_test.py9
1 files changed, 9 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))