Take top 5 instead of top 3
This commit is contained in:
parent
f2d33b51b1
commit
b240317d76
|
@ -2,7 +2,7 @@ import numpy as np
|
|||
import torch
|
||||
|
||||
|
||||
def predict(device, dataset, model, text, next_words=100, top_k=3):
|
||||
def predict(device, dataset, model, text, next_words=100, top_k=5):
|
||||
model.eval()
|
||||
|
||||
words = text.split()
|
||||
|
|
Loading…
Reference in a new issue