Skip to content
Permalink
Browse files

Update utils.py

  • Loading branch information
boudinfl committed Nov 28, 2019
1 parent 03c9028 commit 943899a2fb1fe509d52883e8ca4b2edadd29fe6b
Showing with 1 addition and 1 deletion.
  1. +1 −1 pke/utils.py
@@ -199,7 +199,7 @@ def train_supervised_model(input_dir,
logging.info('reading file {}'.format(input_file))

# get the document id from file name
doc_id = '.'.join(input_file.split('/')[-1].split('.')[0:-1])
doc_id = '.'.join(os.path.basename(input_file).split('.')[0:-1])

# initialize the input file
model.__init__()

0 comments on commit 943899a

Please sign in to comment.
You can’t perform that action at this time.