version 0.1.1

models/node2vec/node2vec.py

@@ -472,14 +472,12 @@ def parse_tensor_f(x):
     data_uri_list = []
     num_rows_saved = 0
     for k, s in enumerate(tqdm(raw_data)):
-        tnow = time.time()
         # generate skipgram
         features = _make_skipgrams(s)
         num_rows_saved += features.shape[0]
 
         data_uri = os.path.join(save_path, f"skipgrams_{k:05}.tfrecord")
         data_uri_list.append(data_uri)
-        t1 = time.time() - tnow
         # Write to tfrecord with proper format
         # tensors2tfrecord(
         #     data_uri,
@@ -488,10 +486,7 @@ def parse_tensor_f(x):
         ds = tf.data.Dataset.from_tensor_slices(features).map(tf.io.serialize_tensor)
         writer = tf.data.experimental.TFRecordWriter(data_uri)
         writer.write(ds)
-
-        t2 = time.time() - tnow  - t1
-        logging.info(f"Making skipgrams batch {k}: generation time: {t1} s, save time: {t2} s")
-
+
     return data_uri_list, num_rows_saved