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
|
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()
|
model.eval()
|
||||||
|
|
||||||
words = text.split()
|
words = text.split()
|
||||||
|
|
Loading…
Reference in a new issue