Class: SVMKit::Preprocessing::LabelEncoder

Inherits:
Object
  • Object
show all
Includes:
Base::BaseEstimator, Base::Transformer
Defined in:
lib/svmkit/preprocessing/label_encoder.rb

Overview

Encode labels to values between 0 and n_classes - 1.

Examples:

encoder = SVMKit::Preprocessing::LabelEncoder.new
labels = Numo::Int32[1, 8, 8, 15, 0]
encoded_labels = encoder.fit_transform(labels)
# > pp encoded_labels
# Numo::Int32#shape=[5]
# [1, 2, 2, 3, 0]
decoded_labels = encoder.inverse_transform(encoded_labels)
# > pp decoded_labels
# [1, 8, 8, 15, 0]

Instance Attribute Summary collapse

Attributes included from Base::BaseEstimator

#params

Instance Method Summary collapse

Constructor Details

#initialize ⇒ LabelEncoder

Create a new encoder for encoding labels to values between 0 and n_classes - 1.



30
31
32
33
 
# File 'lib/svmkit/preprocessing/label_encoder.rb', line 30

def initialize
  @params = {}
  @classes = nil
end

Instance Attribute Details

#classes ⇒ Array (readonly)

Return the class labels.

Returns:

  • (Array) —

    (size: [n_classes])



27
28
29
 
# File 'lib/svmkit/preprocessing/label_encoder.rb', line 27

def classes
  @classes
end

Instance Method Details

#fit(x) ⇒ LabelEncoder

Fit label-encoder to labels.

Parameters:

  • x (Array) —

    (shape: [n_samples]) The labels to fit label-encoder.

Returns:



41
42
43
44
45
46
 
# File 'lib/svmkit/preprocessing/label_encoder.rb', line 41

def fit(x, _y = nil)
  x = x.to_a if x.is_a?(Numo::NArray)
  SVMKit::Validation.check_params_type(Array, x: x)
  @classes = x.sort.uniq
  self
end

#fit_transform(x) ⇒ Numo::Int32

Fit label-encoder to labels, then return encoded labels.

Parameters:

  • x (Array) —

    (shape: [n_samples]) The labels to fit label-encoder.

Returns:

  • (Numo::Int32) —

    The encoded labels.



54
55
56
57
58
 
# File 'lib/svmkit/preprocessing/label_encoder.rb', line 54

def fit_transform(x, _y = nil)
  x = x.to_a if x.is_a?(Numo::NArray)
  SVMKit::Validation.check_params_type(Array, x: x)
  fit(x).transform(x)
end

#inverse_transform(x) ⇒ Array

Decode encoded labels.

Parameters:

  • x (Numo::Int32) —

    (shape: [n_samples]) The labels to be decoded.

Returns:

  • (Array) —

    The decoded labels.



74
75
76
77
 
# File 'lib/svmkit/preprocessing/label_encoder.rb', line 74

def inverse_transform(x)
  SVMKit::Validation.check_label_array(x)
  x.to_a.map { |n| @classes[n] }
end

#marshal_dump ⇒ Hash

Dump marshal data.

Returns:

  • (Hash) —

    The marshal data about LabelEncoder



81
82
83
84
 
# File 'lib/svmkit/preprocessing/label_encoder.rb', line 81

def marshal_dump
  { params: @params,
    classes: @classes }
end

#marshal_load(obj) ⇒ nil

Load marshal data.

Returns:

  • (nil) 


88
89
90
91
92
 
# File 'lib/svmkit/preprocessing/label_encoder.rb', line 88

def marshal_load(obj)
  @params = obj[:params]
  @classes = obj[:classes]
  nil
end

#transform(x) ⇒ Numo::Int32

Encode labels.

Parameters:

  • x (Array) —

    (shape: [n_samples]) The labels to be encoded.

Returns:

  • (Numo::Int32) —

    The encoded labels.



64
65
66
67
68
 
# File 'lib/svmkit/preprocessing/label_encoder.rb', line 64

def transform(x)
  x = x.to_a if x.is_a?(Numo::NArray)
  SVMKit::Validation.check_params_type(Array, x: x)
  Numo::Int32[*(x.map { |v| @classes.index(v) })]
end