#!/usr/bin/perl -w
# construct_dt_and_classify_one_sample_case2.pl
use lib '../blib/lib', '../blib/arch';
## This script is for a mixture of symbolic and numeric features. Since
## we have numeric features, only a csv file can be used for training.
## Note how we tell the module that the class label in each training
## record is placed in the column indexed 2 and how the feature are to be
## found in columns indexed 3, 4, 5, 6, 7, and 8.
## Remember, the column indexing in the csv file is zero-based. That is,
## the first column is indexed 0.
use strict;
use Algorithm::DecisionTree;
#my $training_datafile = "stage3cancer.csv";
my $training_datafile = "stage3cancer_noquotes.csv";
my $dt = Algorithm::DecisionTree->new(
training_datafile => $training_datafile,
csv_class_column_index => 2,
csv_columns_for_features => [3,4,5,6,7,8],
entropy_threshold => 0.01,
max_depth_desired => 8,
symbolic_to_numeric_cardinality_threshold => 10,
);
$dt->get_training_data();
$dt->calculate_first_order_probabilities();
$dt->calculate_class_priors();
# UNCOMMENT THE NEXT STATEMENT if you would like to see the
# training data that was read from the disk file:
#$dt->show_training_data();
my $root_node = $dt->construct_decision_tree_classifier();
# UNCOMMENT THE NEXT TWO STATEMENTs if you would like to see the
# decision tree displayed in your terminal window:
print "\n\nThe Decision Tree:\n\n";
$root_node->display_decision_tree(" ");
### The following test_sample is for the training files with names
### like training.dat training2.dat:
my @test_sample = qw / g2=4.2
grade=2.3
gleason=4
eet=1.7
age=55.0
ploidy=diploid /;
# The classifiy() in the call below returns a reference to a hash
# whose keys are the class labels and the values the associated
# probabilities:
my %classification = %{$dt->classify($root_node, \@test_sample)};
my @solution_path = @{$classification{'solution_path'}};
delete $classification{'solution_path'};
my @which_classes = keys %classification;
@which_classes = sort {$classification{$b} <=> $classification{$a}}
@which_classes;
print "\nClassification:\n\n";
print " class probability\n";
print " ---------- -----------\n";
foreach my $which_class (@which_classes) {
my $classstring = sprintf("%-30s", $which_class);
my $valuestring = sprintf("%-30s", $classification{$which_class});
print " $classstring $valuestring\n";
}
print "\nSolution path in the decision tree: @solution_path\n";
print "\nNumber of nodes created: " . $root_node->how_many_nodes() . "\n";