/StoryEndGen-pytorch

Pytorch implementation of StoryEndGen

Primary LanguagePython

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