Module: DNN::CIFAR10

Defined in:
lib/dnn/lib/cifar10.rb

Class Method Summary collapse

Class Method Details

.dir ⇒ Object 



21
22
23
 
# File 'lib/dnn/lib/cifar10.rb', line 21

def self.dir
  "cifar-10-batches-bin"
end

.load_test ⇒ Object 



16
17
18
19
 
# File 'lib/dnn/lib/cifar10.rb', line 16

def self.load_test
  bin = File.binread("#{dir}/test_batch.bin")
  _cifar10_load(bin, 10000)
end

.load_train ⇒ Object 



8
9
10
11
12
13
14
 
# File 'lib/dnn/lib/cifar10.rb', line 8

def self.load_train
  bin = ""
  (1..5).each do |i|
    bin << File.binread("#{dir}/data_batch_#{i}.bin")
  end
  _cifar10_load(bin, 50000)
end