Используйте функцию предсказания Кераса в сеансе тензорного потока

rand(6)    #=> gives a random number between 0 and 6 inclusively 
rand(1..6) #=> gives a random number between 1 and 6 inclusively

Обратите внимание, что опция диапазона доступна только в новых версиях ruby ​​[1.9+].