dkam/decisiontree
1
0
Fork 0
mirror of https://github.com/dkam/decisiontree.git synced 2025-12-27 22:54:52 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Tidy code style

Browse Source
This commit is contained in:
Sam Oliver
2021-09-17 12:56:27 +02:00
parent e30c18853a
commit aad3b61d10
12 changed files with 205 additions and 204 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
spec/id3_spec.rb
View File

@@ -1,19 +1,18 @@
require 'spec_helper'
require "spec_helper"
describe describe DecisionTree::ID3Tree do
describe "simple discrete case" do
Given(:labels) { ["sun", "rain"]}
Given(:labels) { ["sun", "rain"] }
Given(:data) do
[
[1,0,1],
[0,1,0]
[1, 0, 1],
[0, 1, 0]
]
end
Given(:tree) { DecisionTree::ID3Tree.new(labels, data, 1, :discrete) }
When { tree.train }
Then { expect(tree.predict([1,0])).to eq 1 }
Then { expect(tree.predict([0,1])).to eq 0 }
Then { expect(tree.predict([1, 0])).to eq 1 }
Then { expect(tree.predict([0, 1])).to eq 0 }
end
describe "discrete attributes" do
@@ -84,7 +83,7 @@ describe describe DecisionTree::ID3Tree do
end
Given(:tree) { DecisionTree::ID3Tree.new(labels, data, "RED", :discrete) }
When { tree.train }
Then { expect(tree.predict(["a1","b0","c0"])).to eq "RED" }
Then { expect(tree.predict(["a1", "b0", "c0"])).to eq "RED" }
end
describe "numerical labels case" do
@@ -109,11 +108,11 @@ describe describe DecisionTree::ID3Tree do
File.delete("#{FIGURE_FILENAME}.png") if File.file?("#{FIGURE_FILENAME}.png")
end
Given(:labels) { ["sun", "rain"]}
Given(:labels) { ["sun", "rain"] }
Given(:data) do
[
[1,0,1],
[0,1,0]
[1, 0, 1],
[0, 1, 0]
]
end
Given(:tree) { DecisionTree::ID3Tree.new(labels, data, 1, :discrete) }

6
spec/spec_helper.rb
View File

@@ -1,5 +1,5 @@
require 'rspec/given'
require 'decisiontree'
require 'pry'
require "rspec/given"
require "decisiontree"
require "pry"
FIGURE_FILENAME = "just_a_spec"