Method: Libsvm::Node.features

Defined in:
lib/libsvm/node.rb

.features(*vargs) ⇒ Object



4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
# File 'lib/libsvm/node.rb', line 4

def features(*vargs)
  array_of_nodes = []
  if vargs.size == 1
    if vargs.first.class == Array
      vargs.first.each_with_index do |value, index|
        array_of_nodes << Node.new(index.to_i, value.to_f)
      end
    elsif vargs.first.class == Hash
      vargs.first.each do |index, value|
        array_of_nodes << Node.new(index.to_i, value.to_f)
      end
    else
      raise(ArgumentError.new("Node features need to be a Hash, Array or Floats"))
    end
  else
    vargs.each_with_index do |value, index|
      array_of_nodes << Node.new(index.to_i, value.to_f)
    end
  end
  array_of_nodes
end