ruby-dnn 0.1.2 → 0.1.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 2f509323fa3b0d100943d2f30f9545baf0116e5e9f66c068121712e0b079ad04
-  data.tar.gz: 9e1163fe79527c2514b4cd509d3899e93c9e39e1a3491ce98e9a0d073ee24763
+  metadata.gz: a30ecc0e25621a1ea4c06948c155adb037983241ac8f7f05efa2a28db9e52e48
+  data.tar.gz: 601599d13059ae9db1a4fb8129f3483bbea258a0408c3208b11a4b4f55f82198
 SHA512:
-  metadata.gz: ef8704fb2def4124e0f7d0bfba7fa83ed9c304bf1d366ba3c03fb28b583cc909d969a2991b99fe55c879e6806590e59f76fbc177dc8ddd9643e4d07aef862b07
-  data.tar.gz: b8600ede2ba0e2463a0ca4e1170a3ec94f02a60edf89545927a0eb9752719c48cf13d2b7c4ac38d936bd70fab7d4003df33c82d8625ee827b2aa599572acd466
+  metadata.gz: 6db06348f1a7f7e27a5a0010199302a20000242fba944a03a12aa42ebda7832fa1a80271475de1f20e3866b653e3ebf6dd44c4e42e22bd6c9b1874f6671a4178
+  data.tar.gz: 0af1c1c6bf2d1ed8b48876730a76e3f1324319a075be72885cdcb24ffb0d15de9c673beb6b446b74be5f9298730d9a7e17a4ceb0f2dfe232663b2bd4dcd83824

data/lib/dnn/core/version.rb

@@ -1,3 +1,3 @@
 module DNN
-  VERSION = "0.1.2"
+  VERSION = "0.1.3"
 end

data/lib/dnn/lib/cifar10.rb

@@ -1,26 +1,53 @@
 require "dnn"
 require "dnn/ext/cifar10/cifar10_ext"
+require "open-uri"
+require "zlib"
+require "archive/tar/minitar"
+
+URL_CIFAR10 = "https://www.cs.toronto.edu/~kriz/cifar-10-binary.tar.gz"
+CIFAR10_DIR = "cifar-10-batches-bin"
 
 module DNN
   module CIFAR10
+    class DNN_CIFAR10_LoadError < DNN_Error; end
+
+    class DNN_CIFAR10_DownloadError < DNN_Error; end
+
     private_class_method :_cifar10_load
 
+    def self.downloads
+      return if Dir.exist?(__dir__ + "/" + CIFAR10_DIR)
+      cifar10_binary_file_name = __dir__ + "/" + URL_CIFAR10.match(%r`.+/(.+)`)[1]
+      puts "Now downloading..."
+      open(URL_CIFAR10, "rb") do |f|
+        File.binwrite(cifar10_binary_file_name, f.read)
+        Zlib::GzipReader.open(cifar10_binary_file_name) do |gz|
+          Archive::Tar::Minitar::unpack(gz, __dir__)
+        end
+      end
+      File.delete(cifar10_binary_file_name)
+      puts "The download has ended."
+    rescue => ex
+      raise DNN_CIFAR10_DownloadError.new(ex.message)
+    end
+
     def self.load_train
+      downloads
       bin = ""
       (1..5).each do |i|
-        bin << File.binread("#{dir}/data_batch_#{i}.bin")
+        fname = __dir__ + "/#{CIFAR10_DIR}/data_batch_#{i}.bin"
+        raise DNN_CIFAR10_LoadError.new(%`file "#{fname}" is not found.`) unless File.exist?(fname)
+        bin << File.binread(fname)
       end
       _cifar10_load(bin, 50000)
     end
 
     def self.load_test
-      bin = File.binread("#{dir}/test_batch.bin")
+      downloads
+      fname = __dir__ + "/#{CIFAR10_DIR}/test_batch.bin"
+      raise DNN_CIFAR10_LoadError.new(%`file "#{fname}" is not found.`) unless File.exist?(fname)
+      bin = File.binread(fname)
       _cifar10_load(bin, 10000)
     end
-
-    def self.dir
-      "cifar-10-batches-bin"
-    end
   end
 end
-##30730000

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ruby-dnn
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.1.3
 platform: ruby
 authors:
 - unagiootoro