コード例 #1
0
ファイル: skip_gram_ops_test.py プロジェクト: jasonyum/addons 
 def _make_text_vocab_float_file():
     filepath = os.path.join(test.get_temp_dir(), "vocab_freq_float.txt")
     with open(filepath, "w") as f:
         writer = csv.writer(f)
         writer.writerows([
             ["and", 0.4],
             ["life", 0.08],
             ["the", 0.3],
             ["to", 0.2],
             ["universe", 0.02],
         ])
     return filepath
コード例 #2
0
ファイル: reformer_e2e_test.py プロジェクト: ppvalluri09/trax 
 def tmp_dir(self):
   tmp = tempfile.mkdtemp(dir=test.get_temp_dir())
   yield tmp
   gfile.rmtree(tmp)