samples/iris.rb in tensor_stream-opencl-0.2.1 vs samples/iris.rb in tensor_stream-opencl-0.2.2
- old
+ new
@@ -46,11 +46,9 @@
validation_cases = []
x_test.each_with_index do |x, index|
validation_cases << [x, y_test[index]]
end
-
-
def init_weights(shape)
# Weight initialization
weights = TensorStream.random_normal(shape, stddev: 0.1)
TensorStream.variable(weights)
end
\ No newline at end of file