模型上线

eval_dataset_test.py

@@ -93,6 +93,25 @@ try:
 except:
     print(time.time() - start)
 
+dataset = input_fn(DATA_DIR + 'eval_samples.csv', 1, False, 2 ** 15)
+
+import time
+iter1 = dataset.make_one_shot_iterator()
+try:
+    start = time.time()
+    while True:
+        iter1.get_next()
+except:
+    print(time.time() - start)
+
+
+iter2 = dataset.make_one_shot_iterator()
+try:
+    start = time.time()
+    while True:
+        iter2.get_next()
+except:
+    print(time.time() - start)
 
 iter2 = dataset.make_one_shot_iterator()
 try: