Class: Gliner::Config::ClassificationTask

Inherits:
Object
  • Object
show all
Defined in:
lib/gliner/config/classification_task.rb

Constant Summary collapse

DEFAULT_THRESHOLD = 
0.5

Class Method Summary collapse

Class Method Details

.parse(task_name, config) ⇒ Object 



9
10
11
12
13
14
15
16
17
18
 
# File 'lib/gliner/config/classification_task.rb', line 9

def parse(task_name, config)
  case config
  when Array
    from_labels(config)
  when Hash
    from_hash(task_name, config)
  else
    raise Error, "classification task #{task_name.inspect} must be an Array or Hash"
  end
end