init
This commit is contained in:
10
lib/datasets/test_NLP.py
Normal file
10
lib/datasets/test_NLP.py
Normal file
@@ -0,0 +1,10 @@
|
||||
import os, sys, torch
|
||||
|
||||
from LanguageDataset import SentCorpus, BatchSentLoader
|
||||
|
||||
if __name__ == '__main__':
|
||||
path = '../../data/data/penn'
|
||||
corpus = SentCorpus( path )
|
||||
loader = BatchSentLoader(corpus.test, 10)
|
||||
for i, d in enumerate(loader):
|
||||
print('{:} :: {:}'.format(i, d.size()))
|
Reference in New Issue
Block a user