examples/cifar10_example.rb in ruby-dnn-0.8.6 vs examples/cifar10_example.rb in ruby-dnn-0.8.7
- old
+ new
@@ -15,11 +15,11 @@
x_test = Numo::SFloat.cast(x_test)
x_train /= 255
x_test /= 255
-y_train = DNN::Util.to_categorical(y_train, 10)
-y_test = DNN::Util.to_categorical(y_test, 10)
+y_train = DNN::Util.to_categorical(y_train, 10, Numo::SFloat)
+y_test = DNN::Util.to_categorical(y_test, 10, Numo::SFloat)
model = Model.new
model << InputLayer.new([32, 32, 3])