View difference between Paste ID: vkHJ6AVj and NFafGT6U
SHOW: | | - or go back to the newest paste.
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)