Slide 40
Slide 40 text
decisiontree.gem usage
require 'decisiontree'
# training data (last items are labels)
feature_names = ['hungers', 'color']
examples = [
[8, 'red', 'angry'],
[6, 'red', 'angry'],
[7, 'red', 'angry'],
[7, 'blue', 'not angry'],
[2, 'red', 'not angry'],
[3, 'blue', 'not angry'],
[2, 'blue', 'not angry'],
[1, 'red', 'not angry']
]
# train model
tree = DecisionTree::ID3Tree.new(
feature_names, examples, 'not angry',
color: :descrete, hunger: :continuous
)
tree.train
# prediction
pred = tree.predict([7, 'red', 'angry'])
puts "Predicted: #{pred} for angry"