Module: OnnxRuntime::Datasets

Defined in:
lib/onnxruntime/datasets.rb

Class Method Summary collapse

Class Method Details

.example(name) ⇒ Object



3
4
5
6
7
8
# File 'lib/onnxruntime/datasets.rb', line 3

def self.example(name)
  unless %w(logreg_iris.onnx mul_1.onnx sigmoid.onnx).include?(name)
    raise ArgumentError, "Unable to find example '#{name}'"
  end
  File.expand_path("../../datasets/#{name}", __dir__)
end