Lyusungwon / StoryEndGen-pytorch

Pytorch implementation of StoryEndGen

StoryEndGen-pytorch

Pytorch implementation of StoryEndGen. Data preprocessed from https://github.com/JianGuanTHU/StoryEndGen.git.

Dependency

• Pattern

  • Install Pattern through PyPi repository

    pip install pattern

  • To handle Pattern-3.6 Issue with python 3.7, replace the following lines 608-609 of [site-package dir] /Pattern-3.6-py3.7.egg/pattern/text/__init__.py

            yield line
      raise StopIteration

    with

      try:
          yield line
      except:
          StopIteration