Sha256: 78bc7ca438087ca2fc318f7f711ee4e565c6df08091145d61c99e9e01fc9aa1f

Contents?: true

Size: 669 Bytes

Versions: 3

Compression:

Stored size: 669 Bytes

Contents

require File.join(File.dirname(__FILE__), "/../spec_helper")
require 'data_frame/id3'

describe "DecisionTree" do

  before do
    @filename = File.expand_path(File.join(File.dirname(__FILE__), "../fixtures/discrete_training.csv"))
    @df = DataFrame.from_csv(@filename)
    @test_data = File.read(@filename)
  end
  
  it "should require the decisiontree gem" do
    defined?(DecisionTree::ID3Tree).should eql('constant')
  end
  
  it "should be able to create a decision tree from a data frame" do
    # Come back to this.
    # @df.create_id3(:purchase)
    # @df.id3.train
    # @df.id3.predict(["36 - 55", "masters", "high", "single", 1]).should eql(1)
  end
end

Version data entries

3 entries across 3 versions & 2 rubygems

Version Path
davidrichards-data_frame-0.0.19 spec/data_frame/id3_spec.rb
davidrichards-data_frame-0.0.20 spec/data_frame/id3_spec.rb
data_frame-0.1.8 spec/data_frame/id3_spec.rb