Advertisement
Guest User

play.py set temp and top_k

a guest
Dec 11th, 2019
352
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 1.69 KB | None | 0 0
  1. # ORIGINAL CODE
  2. def play_aidungeon_2():
  3.  
  4.     console_print("AI Dungeon 2 will save and use your actions and game to continually improve AI Dungeon."
  5.                   + " If you would like to disable this enter 'nosaving' for any action. This will also turn off the "
  6.                   + "ability to save games.")
  7.  
  8.     upload_story = True
  9.  
  10.     print("\nInitializing AI Dungeon! (This might take a few minutes)\n")
  11.     generator = GPT2Generator()
  12.  
  13. # NEW CODE
  14. def play_aidungeon_2():
  15.  
  16.     console_print(
  17.         "\nAI Dungeon 2 will save and use your actions and game to continually improve AI Dungeon."
  18.         + " If you would like to disable this enter 'nosaving' for any action. This will also turn off the "
  19.         + "ability to save games.\n"
  20.     )
  21.  
  22.     upload_story = True
  23.  
  24.     # "temperature" dictates randomness. A low temperature means that the AI is
  25.     #  more likely to go with the word that best fits the context, a high
  26.     #  temperature makes the AI more random and it may chose surprising less fitting words
  27.     # original 0.4
  28.     temp = 0.2
  29.  
  30.     # lower top_k is a hard limit of "how many fitting words should I consider",
  31.     #  i.e. lowering this value also limits the AI in creativity
  32.     # original 40
  33.     top_k = 20
  34.  
  35.     console_print("\nBefore we start, would you like to change the default temperature (" + str(temp) + ") and top_k (" + str(top_k) + ") value?\n")
  36.     choice = input("1) Aeeyup!\n-) Press enter to skip\n>")
  37.  
  38.     if choice == "1":
  39.         temp = float(input("New temperature: "))
  40.         top_k = int(input("New top_k: "))
  41.  
  42.     print("\nInitializing AI Dungeon! (This might take a few minutes)\n")
  43.     generator = GPT2Generator(60, temp, top_k, 0.9)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement