romman8's picture
add: predictions on test set
13b6847