You are on page 1of 128

Data Mining

Classification: Basic Concepts, Decision


Trees, and Model Evaluation

Lecture Notes for Chapter 4

Introduction to Data Mining


by
Tan, Steinbach, Kumar

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 1


Classification: Definition

 Given a collection of records (training set )


– Each record contains a set of attributes, one of the
attributes is the class.
 Find a model for class attribute as a function
of the values of other attributes.
 Goal: previously unseen records should be
assigned a class as accurately as possible.
– A test set is used to determine the accuracy of the
model. Usually, the given data set is divided into
training and test sets, with training set used to build
the model and test set used to validate it.

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 2


Classification: Definition

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 3


Illustrating Classification Task

Tid Attrib1 Attrib2 Attrib3 Class Learning


1 Yes Large 125K No
algorithm
2 No Medium 100K No

3 No Small 70K No

4 Yes Medium 120K No


Induction
5 No Large 95K Yes

6 No Medium 60K No

7 Yes Large 220K No Learn


8 No Small 85K Yes Model
9 No Medium 75K No

10 No Small 90K Yes


Model
10

Training Set
Apply
Tid Attrib1 Attrib2 Attrib3 Class Model
11 No Small 55K ?

12 Yes Medium 80K ?

13 Yes Large 110K ? Deduction


14 No Small 95K ?

15 No Large 67K ?
10

Test Set

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 4


Examples of Classification Task

 Predicting tumor cells as benign or malignant

 Classifying credit card transactions


as legitimate or fraudulent

 Classifying secondary structures of protein


as alpha-helix, beta-sheet, or random
coil

 Categorizing news stories as finance,


weather, entertainment, sports, etc
© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 5
Classification Techniques

 Decision Tree based Methods


 Rule-based Methods
 Memory based reasoning
 Neural Networks
 Naïve Bayes and Bayesian Belief Networks
 Support Vector Machines

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 6


Example of a Decision Tree

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 7


Example of a Decision Tree

Splitting Attributes
Tid Refund Marital Taxable
Status Income Cheat

1 Yes Single 125K No


2 No Married 100K No Refund
No
Yes No
3 No Single 70K
4 Yes Married 120K No NO MarSt
5 No Divorced 95K Yes Single, Divorced Married
6 No Married 60K No
7 Yes Divorced 220K No TaxInc NO
8 No Single 85K Yes < 80K > 80K
9 No Married 75K No
NO YES
10 No Single 90K Yes
10

Training Data Model: Decision Tree

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 8


Another Example of Decision Tree

MarSt Single,
Married Divorced
Tid Refund Marital Taxable
Status Income Cheat
NO Refund
1 Yes Single 125K No
Yes No
2 No Married 100K No
3 No Single 70K No NO TaxInc
4 Yes Married 120K No < 80K > 80K
5 No Divorced 95K Yes
NO YES
6 No Married 60K No
7 Yes Divorced 220K No
8 No Single 85K Yes
9 No Married 75K No There could be more than one tree that
10 No Single 90K Yes fits the same data!
10

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 9


Decision Tree Classification Task

Tid Attrib1 Attrib2 Attrib3 Class


Tree
1 Yes Large 125K No Induction
2 No Medium 100K No algorithm
3 No Small 70K No

4 Yes Medium 120K No


Induction
5 No Large 95K Yes

6 No Medium 60K No

7 Yes Large 220K No Learn


8 No Small 85K Yes Model
9 No Medium 75K No

10 No Small 90K Yes


Model
10

Training Set
Apply Decision
Tid Attrib1 Attrib2 Attrib3 Class
Model Tree
11 No Small 55K ?

12 Yes Medium 80K ?

13 Yes Large 110K ?


Deduction
14 No Small 95K ?

15 No Large 67K ?
10

Test Set

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 10


Apply Model to Test Data

Test Data
Start from the root of tree. Refund Marital Taxable
Status Income Cheat

No Married 80K ?
Refund 10

Yes No

NO MarSt
Single, Divorced Married

TaxInc NO
< 80K > 80K

NO YES

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 11


Apply Model to Test Data

Test Data
Refund Marital Taxable
Status Income Cheat

No Married 80K ?
Refund 10

Yes No

NO MarSt
Single, Divorced Married

TaxInc NO
< 80K > 80K

NO YES

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 12


Apply Model to Test Data

Test Data
Refund Marital Taxable
Status Income Cheat

No Married 80K ?
Refund 10

Yes No

NO MarSt
Single, Divorced Married

TaxInc NO
< 80K > 80K

NO YES

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 13


Apply Model to Test Data

Test Data
Refund Marital Taxable
Status Income Cheat

No Married 80K ?
Refund 10

Yes No

NO MarSt
Single, Divorced Married

TaxInc NO
< 80K > 80K

NO YES

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 14


Apply Model to Test Data

Test Data
Refund Marital Taxable
Status Income Cheat

No Married 80K ?
Refund 10

Yes No

NO MarSt
Single, Divorced Married

TaxInc NO
< 80K > 80K

NO YES

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 15


Apply Model to Test Data

Test Data
Refund Marital Taxable
Status Income Cheat

No Married 80K ?
Refund 10

Yes No

NO MarSt
Single, Divorced Married Assign Cheat to “No”

TaxInc NO
< 80K > 80K

NO YES

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 16


Decision Tree Classification Task

Tid Attrib1 Attrib2 Attrib3 Class


Tree
1 Yes Large 125K No Induction
2 No Medium 100K No algorithm
3 No Small 70K No

4 Yes Medium 120K No


Induction
5 No Large 95K Yes

6 No Medium 60K No

7 Yes Large 220K No Learn


8 No Small 85K Yes Model
9 No Medium 75K No

10 No Small 90K Yes


Model
10

Training Set
Apply Decision
Tid Attrib1 Attrib2 Attrib3 Class
Model Tree
11 No Small 55K ?

12 Yes Medium 80K ?

13 Yes Large 110K ?


Deduction
14 No Small 95K ?

15 No Large 67K ?
10

Test Set

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 17


Decision Tree Induction

 Many Algorithms:
– Hunt’s Algorithm (one of the earliest)
– CART
– ID3, C4.5
– SLIQ,SPRINT

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 18


General Structure of Hunt’s Algorithm
Tid Refund Marital Taxable
 Let Dt be the set of training records Status Income Cheat
that reach a node t 1 Yes Single 125K No

 General Procedure: 2 No Married 100K No

– If Dt contains records that 3 No Single 70K No


4 Yes Married 120K No
belong the same class yt, then t
5 No Divorced 95K Yes
is a leaf node labeled as yt 6 No Married 60K No
– If Dt is an empty set, then t is a 7 Yes Divorced 220K No
leaf node labeled by the default 8 No Single 85K Yes
class, yd 9 No Married 75K No

– If Dt contains records that 10


10 No Single 90K Yes

belong to more than one class, Dt


use an attribute test to split the
data into smaller subsets.
Recursively apply the ?
procedure to each subset.

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 19


Hunt’s Algorithm Tid Refund Marital
Status
Taxable
Income Cheat

1 Yes Single 125K No


2 No Married 100K No
Refund
Don’t 3 No Single 70K No
Yes No
Cheat 4 Yes Married 120K No
Don’t Don’t 5 No Divorced 95K Yes
Cheat Cheat
6 No Married 60K No
7 Yes Divorced 220K No
8 No Single 85K Yes

Refund Refund 9 No Married 75K No


Yes No Yes No 10 No Single 90K Yes
10

Don’t Don’t Marital


Marital Cheat
Cheat Status Status
Single, Single,
Married Married
Divorced Divorced

Don’t Taxable Don’t


Cheat Cheat
Cheat Income
< 80K >= 80K

Don’t Cheat
Cheat
© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 20
Tree Induction

 Greedy strategy.
– Split the records based on an attribute test
that optimizes certain criterion.

 Issues
– Determine how to split the records
How to specify the attribute test condition?
How to determine the best split?

– Determine when to stop splitting

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 21


Tree Induction

 Greedy strategy.
– Split the records based on an attribute test
that optimizes certain criterion.

 Design Issues
– Determine how to split the records
How to specify the attribute test condition?
How to determine the best split?

– Determine when to stop splitting

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 22


How to Specify Test Condition?

 Depends on attribute types


– Nominal
– Ordinal
– Continuous

 Depends on number of ways to split


– 2-way split
– Multi-way split

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 23


Splitting Based on Nominal Attributes

 Multi-way split: Use as many partitions as distinct


values.
CarType
Family Luxury
Sports

 Binary split: Divides values into two subsets.


Need to find optimal partitioning.
CarType CarType
{Sports, OR {Family,
Luxury} {Family} Luxury} {Sports}

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 24


Splitting Based on Ordinal Attributes

 Multi-way split: Use as many partitions as distinct


values.
Size
Small Large
Medium

 Binary split: Divides values into two subsets.


Need to find optimal partitioning.
Size Size
{Small,
{Large}
OR {Medium,
{Small}
Medium} Large}

Size
{Small,
 What about this split? Large} {Medium}

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 25


Splitting Based on Continuous Attributes

 Different ways of handling


– Discretization to form an ordinal categorical
attribute
 Static – discretize once at the beginning
 Dynamic – ranges can be found by equal interval
bucketing, equal frequency bucketing
(percentiles), or clustering.

– Binary Decision: (A < v) or (A  v)


 consider all possible splits and finds the best cut
 can be more compute intensive

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 26


Splitting Based on Continuous Attributes

Taxable Taxable
Income Income?
> 80K?
< 10K > 80K
Yes No

[10K,25K) [25K,50K) [50K,80K)

(i) Binary split (ii) Multi-way split

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 27


Tree Induction

 Greedy strategy.
– Split the records based on an attribute test
that optimizes certain criterion.

 Issues
– Determine how to split the records
How to specify the attribute test condition?
How to determine the best split?

– Determine when to stop splitting

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 28


How to determine the Best Split

Before Splitting: 10 records of class 0,


10 records of class 1

Own Car Student


Car? Type? ID?

Yes No Family Luxury c1 c20


c10 c11
Sports
C0: 6 C0: 4 C0: 1 C0: 8 C0: 1 C0: 1 ... C0: 1 C0: 0 ... C0: 0
C1: 4 C1: 6 C1: 3 C1: 0 C1: 7 C1: 0 C1: 0 C1: 1 C1: 1

Which test condition is the best?

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 29


How to determine the Best Split

 Greedy approach:
– Nodes with homogeneous class distribution
are preferred
 Need a measure of node impurity:

C0: 5 C0: 9
C1: 5 C1: 1

Non-homogeneous, Homogeneous,
High degree of impurity Low degree of impurity

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 30


Measures of Node Impurity

 Gini Index
 Entropy
 Misclassification error

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 31


How to Find the Best Split
Before Splitting: C0 N00
M0
C1 N01

A? B?
Yes No Yes No

Node N1 Node N2 Node N3 Node N4

C0 N10 C0 N20 C0 N30 C0 N40


C1 N11 C1 N21 C1 N31 C1 N41

M1 M2 M3 M4

M12 M34
Gain = M0 – M12 vs M0 – M34
© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 32
Measure of Impurity: GINI

 Gini Index for a given node t :

GINI (t )  1  [ p( j | t )]2
j

(NOTE: p( j | t) is the relative frequency of class j at node t).

– Maximum (1 - 1/nc) when records are equally


distributed among all classes, implying least
interesting information
– Minimum (0.0) when all records belong to one class,
implying most interesting information
C1 0 C1 1 C1 2 C1 3
C2 6 C2 5 C2 4 C2 3
Gini=0.000 Gini=0.278 Gini=0.444 Gini=0.500

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 33


Examples for computing GINI

GINI (t )  1  [ p( j | t )]2
j

C1 0 P(C1) = 0/6 = 0 P(C2) = 6/6 = 1


C2 6 Gini = 1 – P(C1)2 – P(C2)2 = 1 – 0 – 1 = 0

C1 1 P(C1) = 1/6 P(C2) = 5/6


C2 5 Gini = 1 – (1/6)2 – (5/6)2 = 0.278

C1 2 P(C1) = 2/6 P(C2) = 4/6


C2 4 Gini = 1 – (2/6)2 – (4/6)2 = 0.444

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 34


Splitting Based on GINI

 Used in CART, SLIQ, SPRINT.


 When a node p is split into k partitions (children), the
quality of split is computed as,
k
ni
GINI split   GINI (i)
i 1 n

where, ni = number of records at child i,


n = number of records at node p.

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 35


Binary Attributes: Computing GINI Index

 Splits into two partitions Parent


C1 6
 Effect of Weighing partitions:
C2 6
– Larger and Purer Partitions are sought for.
Gini = 0.500

A? B?
Yes No Yes No
Node N1 Node N2 Node N1 Node N2

Gini(N1) = N1 N2 Gini(N1) = N1 N2
1– (5/6)2 – (2/6)2 = 0.194
1 – (4/6)2 – (3/6)2 = 0.489 C1 4 2 C1 5 1
Gini(N2) =
Gini(N2) =
C2 3 3 1 – (1/6)2 – (4/6)2 = 0.528
C2 2 4
1 – (2/6)2 – (3/6)2 = 0.480
Gini=0.486 Gini=0.333
Gini(Children) Gini(Children)
= 7/12 * 0.489 + 5/12 * 0.480 = 0.486 = 7/12 * 0.194 + 5/12 * 0.528 = 0.333

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 36


Categorical Attributes: Computing Gini Index

 For each distinct value, gather counts for each class in


the dataset
 Use the count matrix to make decisions

Multi-way split Two-way split


(find best partition of values)

CarType CarType CarType


Family Sports Luxury {Sports, {Family,
{Family} {Sports}
Luxury} Luxury}
C1 1 2 1 C1 3 1 C1 2 2
C2 4 1 1 C2 2 4 C2 1 5
Gini 0.393 Gini 0.400 Gini 0.419

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 37


Continuous Attributes: Computing Gini Index

Tid Refund Marital Taxable


 Use Binary Decisions based on one Status Income Cheat
value
1 Yes Single 125K No
 Several Choices for the splitting value 2 No Married 100K No
– Number of possible splitting values 3 No Single 70K No
= Number of distinct values 4 Yes Married 120K No

 Each splitting value has a count matrix 5 No Divorced 95K Yes

associated with it 6 No Married 60K No


7 Yes Divorced 220K No
– Class counts in each of the
partitions, A < v and A  v 8 No Single 85K Yes
9 No Married 75K No
 Simple method to choose best v 10 No Single 90K Yes
– For each v, scan the database to
10

gather count matrix and compute Taxable


its Gini index Income
> 80K?
– Computationally Inefficient!
Repetition of work. Yes No

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 38


Continuous Attributes: Computing Gini Index...

 For efficient computation: for each attribute,


– Sort the attribute on values
– Linearly scan these values, each time updating the count matrix
and computing gini index
– Choose the split position that has the least gini index

Cheat No No No Yes Yes Yes No No No No


Taxable Income

Sorted Values 60 70 75 85 90 95 100 120 125 220


55 65 72 80 87 92 97 110 122 172 230
Split Positions
<= > <= > <= > <= > <= > <= > <= > <= > <= > <= > <= >
Yes 0 3 0 3 0 3 0 3 1 2 2 1 3 0 3 0 3 0 3 0 3 0

No 0 7 1 6 2 5 3 4 3 4 3 4 3 4 4 3 5 2 6 1 7 0

Gini 0.420 0.400 0.375 0.343 0.417 0.400 0.300 0.343 0.375 0.400 0.420

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 39


Alternative Splitting Criteria based on INFO

 Entropy at a given node t:


Entropy(t )   p( j | t ) log p( j | t )
j

(NOTE: p( j | t) is the relative frequency of class j at node t).


– Measures homogeneity of a node.
 Maximum (log nc) when records are equally distributed
among all classes implying least information
 Minimum (0.0) when all records belong to one class,
implying most information
– Entropy based computations are similar to the
GINI index computations
© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 40
Examples for computing Entropy

Entropy(t )   p( j | t ) log p( j | t )
j 2

C1 0 P(C1) = 0/6 = 0 P(C2) = 6/6 = 1


C2 6 Entropy = – 0 log 0 – 1 log 1 = – 0 – 0 = 0

C1 1 P(C1) = 1/6 P(C2) = 5/6


C2 5 Entropy = – (1/6) log2 (1/6) – (5/6) log2 (1/6) = 0.65

C1 2 P(C1) = 2/6 P(C2) = 4/6


C2 4 Entropy = – (2/6) log2 (2/6) – (4/6) log2 (4/6) = 0.92

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 41


Splitting Based on INFO...

 Information Gain:
 n 
 Entropy( p)  
k
GAIN Entropy(i )  i

 n 
split i 1

Parent Node, p is split into k partitions;


ni is number of records in partition i
– Measures Reduction in Entropy achieved because of
the split. Choose the split that achieves most reduction
(maximizes GAIN)
– Used in ID3 and C4.5
– Disadvantage: Tends to prefer splits that result in large
number of partitions, each being small but pure.

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 42


Splitting Based on INFO...

 Gain Ratio:
GAIN n n
 SplitINFO   log
k
GainRATIO split
Split i i

SplitINFO n n i 1

Parent Node, p is split into k partitions


ni is the number of records in partition i

– Adjusts Information Gain by the entropy of the


partitioning (SplitINFO). Higher entropy partitioning
(large number of small partitions) is penalized!
– Used in C4.5
– Designed to overcome the disadvantage of Information
Gain
© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 43
Splitting Criteria based on Classification Error

 Classification error at a node t :

Error(t )  1  max P(i | t ) i

 Measures misclassification error made by a node.


 Maximum (1 - 1/nc) when records are equally distributed
among all classes, implying least interesting information
 Minimum (0.0) when all records belong to one class, implying
most interesting information

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 44


Examples for Computing Error

Error(t )  1  max P(i | t ) i

C1 0 P(C1) = 0/6 = 0 P(C2) = 6/6 = 1


C2 6 Error = 1 – max (0, 1) = 1 – 1 = 0

C1 1 P(C1) = 1/6 P(C2) = 5/6


C2 5 Error = 1 – max (1/6, 5/6) = 1 – 5/6 = 1/6

C1 2 P(C1) = 2/6 P(C2) = 4/6


C2 4 Error = 1 – max (2/6, 4/6) = 1 – 4/6 = 1/3

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 45


Comparison among Splitting Criteria

For a 2-class problem:

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 46


Misclassification Error vs Gini

A? Parent
C1 7
Yes No
C2 3
Node N1 Node N2 Gini = 0.42

Gini(N1) N1 N2
= 1 – (3/3)2 – (0/3)2 Gini(Children)
C1 3 4 = 3/10 * 0
=0
C2 0 3 + 7/10 * 0.489
Gini(N2) Gini=0.361 = 0.342
= 1 – (4/7)2 – (3/7)2
= 0.489 Gini improves !!

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 47


Tree Induction

 Greedy strategy.
– Split the records based on an attribute test
that optimizes certain criterion.

 Issues
– Determine how to split the records
How to specify the attribute test condition?
How to determine the best split?

– Determine when to stop splitting

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 48


Stopping Criteria for Tree Induction

 Stop expanding a node when all the records


belong to the same class

 Stop expanding a node when all the records have


similar attribute values

 Early termination (to be discussed later)

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 49


Decision Tree Based Classification

 Advantages:
– Inexpensive to construct
– Extremely fast at classifying unknown records
– Easy to interpret for small-sized trees
– Accuracy is comparable to other classification
techniques for many simple data sets

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 50


Decision Tree Induction Algorithm

function extends the decision tree by


creating a newdetermines
function node. A node the in thelabel to
class
decision tree has to
be assigned either
a a test
condition, denoted as node.test cond,
leaf node.
or a class label, is
function
function denoted
used toasterminate
determines which attribute
the
node.label.
should be selected
tree-growing processas by
thetesting
test
conditionallfor
whether thesplitting
recordsthehavetraining
either
records.
the same class label or the same
attribute values. Another way to
terminate the recursive function is to
test whether the number of records
have fallen below some minimum
threshold.

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 51


Example of a Web Server log

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 52


Web Robot Detection

 2916 records
– 10% for training
– 90% for testing
 Human user: Class 0
 Web Robot: Class 1

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 53


Example

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 54


Example

 Model Attribute can be tossed out, since its


always unique, and it doesn’t help our result.

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 55


Example

 Establish a target classification


 Is the car fast?
 6/15 yes, 9/15 no

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 56


Example – Classification Entropy

 Calculating for the Classification Entropy

 IE= -(6/15)log2(6/15)-(9/15)log2(9/15) = ~0.971

 Must calculate Information Gain of remaining


attributes to determine the root node.

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 57


Example – Information Gain

 Engine: 6 small, 5 medium, 4 large


 3 values for attribute engine, so we need 3 entropy calculations

small: 5 no, 1 yes Ismall = -(5/6)log2(5/6)-(1/6)log2(1/6) = ~0.65

medium: 3 no, 2 yes Imedium = -(3/5)log2(3/5)-(2/5)log2(2/5) = ~0.97

large: 2 no, 2 yes Ilarge = 1 (evenly distributed subset)

IGEngine = IE(S) – [(6/15)*Ismall + (5/15)*Imedium + (4/15)*Ilarge]

IGEngine = 0.971 – 0.85 = 0.121

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 58


Example – Information Gain

 SC/Turbo: 4 yes, 11 no
 2 values for attribute SC/Turbo, so we need 2 entropy calculations

yes: 2 yes, 2 no Iturbo = 1 (evenly distributed subset)

no: 3 yes, 8 no Inoturbo = -(3/11)log2(3/11)-(8/11)log2(8/11) = ~0.84

IGturbo = IE(S) – [(4/15)*Iturbo + (11/15)*Inoturbo]

IGturbo = 0.971 – 0.886 = 0.085

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 59


Example – Information Gain

 Weight: 6 Average, 4 Light, 5 Heavy


 3 values for attribute weight, so we need 3 entropy calculations

average: 3 no, 3 yes Iaverage = 1 (evenly distributed subset)

light: 3 no, 1 yes Ilight = -(3/4)log2(3/4)-(1/4)log2(1/4) = ~0.81

heavy: 4 no, 1 yes Iheavy = -(4/5)log2(4/5)-(1/5)log2(1/5) = ~0.72

IGWeight = IE(S) – [(6/15)*Iaverage + (4/15)*Ilight + (5/15)*Iheavy]

IGWeight = 0.971 – 0.856 = 0.115

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 60


Example – Information Gain

 Fuel Economy: 2 good, 3 average, 10 bad


 3 values for attribute Fuel Eco, so we need 3 entropy calculations

good: 0 yes, 2 no Igood = 0 (no variability)

average: 0 yes, 3 no Iaverage = 0 (no variability)

bad: 5 yes, 5 no Ibad = 1 (evenly distributed subset)

We can omit calculations for good and average since they always end up not fast.

IGFuelEco = IE(S) – [(10/15)*Ibad]


IGFuelEco = 0.971 – 0.667 = 0.304

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 61


Example – Choosing the Root Node

 Recap:
IGEngine 0.121
IGturbo 0.085
IGWeight 0.115
IGFuelEco 0.304
Our best pick is Fuel Eco, and we can immediately predict the car is not fast
when fuel economy is good or average.

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 62


Example – Root of Decision Tree

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 63


Example – After Root Node Creation

 Since we selected the Fuel Eco attribute for our Root Node, it is
removed from the table for future calculations.

Calculating for Entropy IE(Fuel Eco) we get 1, since we have 5 yes and 5 no.

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 64


Example – Information Gain

 Engine: 1 small, 5 medium, 4 large


 3 values for attribute engine, so we need 3 entropy calculations

small: 1 yes, 0 no Ismall = 0 (no variability)

medium: 2 yes, 3 no Imedium = -(2/5)log2(2/5)-(3/5)log2(3/5) = ~0.97

large: 2 no, 2 yes Ilarge = 1 (evenly distributed subset)

IGEngine = IE(SFuelEco) – (5/10)*Imedium + (4/10)*Ilarge]

IGEngine = 1 – 0.885 = 0.115

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 65


Example – Information Gain

 SC/Turbo: 3 yes, 7 no
 2 values for attribute SC/Turbo, so we need 2 entropy calculations

yes: 2 yes, 1 no Iturbo = -(2/3)log2(2/3)-(1/3)log2(1/3) = ~0.84

no: 3 yes, 4 no Inoturbo = -(3/7)log2(3/7)-(4/7)log2(4/7) = ~0.84

IGturbo = IE(SFuelEco) – [(3/10)*Iturbo + (7/10)*Inoturbo]

IGturbo = 1 – 0.965 = 0.035

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 66


Example – Information Gain

 Weight: 3 average, 5 heavy, 2 light


 3 values for attribute weight, so we need 3 entropy calculations

average: 3 yes, 0 no Iaverage = 0 (no variability)

heavy: 1 yes, 4 no Iheavy = -(1/5)log2(1/5)-(4/5)log2(4/5) = ~0.72

light: 1 yes, 1 no Ilight = 1 (evenly distributed subset)

IGEngine = IE(SFuel Eco) – [(5/10)*Iheavy+(2/10)*Ilight]

IGEngine = 1 – 0.561 = 0.439

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 67


Example – Choosing the Level 2 Node

 Recap:
IGEngine 0.115
IGturbo 0.035
IGWeight 0.439

Weight has the highest gain, and is thus the best choice.

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 68


Example – Decision Tree

Since there are only two items for


SC/Turbo where Weight = Light, and
the result is consistent, we can
simplify the weight = Light path.

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 69


Example – Updated Table

All cars with large engines in this table are not fast.

Due to inconsistent patterns in the data, there is no way to proceed


since medium size engines may lead to either fast or not fast.

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 70


Practical Issues of Classification

 Underfitting and Overfitting

 Missing Values

 Costs of Classification

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 72


Underfitting and Overfitting (Example)

500 circular and 500


triangular data points.

Circular points:
0.5  sqrt(x12+x22)  1

Triangular points:
sqrt(x12+x22) > 0.5 or
sqrt(x12+x22) < 1

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 73


Underfitting and Overfitting

Overfitting

Underfitting: when model is too simple, both training and test errors are large
© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 74
Underfitting and Overfitting

 Two decision trees with different number of nodes.


 The tree that contains the smaller number of nodes has a higher
training error rate, but a lower test error rate compared to the more
complex tree

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 75


Overfitting due to Noise

Decision boundary is distorted by noise point

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 76


Overfitting due to Noise

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 77


Overfitting due to Noise

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 78


Overfitting due to Noise

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 79


Overfitting due to Insufficient Examples
Models that make their classification
decisions based on a small number of
training records are also susceptible to
overfitting

Lack of data points in the lower half of the diagram makes it difficult
to predict correctly the class labels of that region
- Insufficient number of training records in the region causes the
decision tree to predict the test examples using other training
records that are irrelevant to the classification task
© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 80
Overfitting due to Insufficient Examples

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 81


Notes on Overfitting

 Overfitting results in decision trees that are more


complex than necessary

 Training error no longer provides a good estimate


of how well the tree will perform on previously
unseen records

 Need new ways for estimating errors

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 82


Estimating Generalization Errors

 1- Using Re-substitution errors: error on training ( e(t) )

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 83


Estimating Generalization Errors

2- Incorporating Model Complexity


 Definition: Occam’s Razor
– Given two models of similar generalization
errors, one should prefer the simpler model
over the more complex model.
– For complex models, there is a greater
chance that it was fitted accidentally by errors
in data.
– Therefore, one should include model
complexity when evaluating a model.

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 85


Incorporating Model Complexity

 Two methods for incorporating model complexity


into the evaluation of classification models
– Pessimistic Error Estimate.
– Minimum Description Length (MDL)

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 86


Pessimistic Error Estimate

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 87


Minimum Description Length (MDL)
A?
X y Yes No
X y
X1 1 0 B? X1 ?
X2 0 B1 B2
X2 ?
X3 0 C? 1
A C1 C2 B X3 ?
X4 1
0 1 X4 ?
… …
Xn
… …
1
Xn ?
 Cost(Model,Data) = Cost(Model) + Cost(Data|Model)
– Cost is the number of bits needed for encoding.
– Search for the least costly model.
 where the first term on the right-hand side is the cost of encoding the
model, while the second term represents the cost of encoding the
mislabeled records. According to the MDL principle, we should seek
a model that minimizes the overall cost function.

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 88


Minimum Description Length (MDL)

 data set that contains 16 binary


attributes and 3 classes, C1, C2,
and C3.
 Cost(Model,Data) = Cost(Model) +
Cost(Data|Model)

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 89


How to Address Overfitting

 Pre-Pruning (Early Stopping Rule)


– Stop the algorithm before it becomes a fully-grown tree
– Typical stopping conditions for a node:
 Stop if all instances belong to the same class
 Stop if all the attribute values are the same
– More restrictive conditions:
 Stop if number of instances is less than some user-specified
threshold
 Stop if class distribution of instances are independent of the
available features (e.g., using  2 test)
 Stop if expanding the current node does not improve impurity
measures (e.g., Gini or information gain).

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 90


How to Address Overfitting…

 Post-pruning
– Grow decision tree to its entirety
– Trim the nodes of the decision tree in a
bottom-up fashion
– If generalization error improves after trimming,
replace sub-tree by a leaf node.
– Class label of leaf node is determined from
majority class of instances in the sub-tree
– Can use MDL for post-pruning

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 91


Handling Missing Attribute Values

 Missing values affect decision tree construction in


three different ways:
– Affects how impurity measures are computed
– Affects how to distribute instance with missing
value to child nodes
– Affects how a test instance with missing value
is classified

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 94


Computing Impurity Measure

Tid Refund Marital Taxable Before Splitting:


Status Income Class Entropy(Parent)
= -0.3 log(0.3)-(0.7)log(0.7) = 0.8813
1 Yes Single 125K No
2 No Married 100K No Class Class
3 No Single 70K No = Yes = No
Refund=Yes 0 3
4 Yes Married 120K No
Refund=No 2 4
5 No Divorced 95K Yes
Refund=? 1 0
6 No Married 60K No
7 Yes Divorced 220K No
Split on Refund:

8 No Single 85K Yes Entropy(Refund=Yes) = 0


9 No Married 75K No Entropy(Refund=No)
10 ? Single 90K Yes = -(2/6)log(2/6) – (4/6)log(4/6) = 0.9183
10

Entropy(Children)
Missing = 0.3 (0) + 0.6 (0.9183) = 0.551
value
Gain = 0.9  (0.8813 – 0.551) = 0.3303
© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 95
Distribute Instances

Tid Refund Marital Taxable


Status Income Class
Tid Refund Marital Taxable
1 Yes Single 125K No Status Income Class
2 No Married 100K No
10 ? Single 90K Yes
3 No Single 70K No 10

4 Yes Married 120K No


Refund
5 No Divorced 95K Yes Yes No
6 No Married 60K No
Class=Yes 0 + 3/9 Class=Yes 2 + 6/9
7 Yes Divorced 220K No
Class=No 3 Class=No 4
8 No Single 85K Yes
9 No Married 75K No
10

Probability that Refund=Yes is 3/9


Refund
Yes No Probability that Refund=No is 6/9
Assign record to the left child with
Class=Yes 0 Cheat=Yes 2 weight = 3/9 and to the right child
Class=No 3 Cheat=No 4 with weight = 6/9

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 96


Classify Instances

New record: Married Single Divorced Total


Tid Refund Marital Taxable
Status Income Class Class=No 3 1 0 4
11 No ? 85K ?
10

Class=Yes 6/9 1 1 2.67

Total 3.67 2 1 6.67


Refund
Yes
No
NO MarSt
Single,
Married Probability that Marital Status
Divorced
= Married is 3.67/6.67
TaxInc NO
< 80K > 80K Probability that Marital Status
={Single,Divorced} is 3/6.67
NO YES

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 97


Other Issues

 Data Fragmentation
 Search Strategy
 Expressiveness
 Tree Replication

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 98


Data Fragmentation

 Number of instances gets smaller as you traverse


down the tree

 Number of instances at the leaf nodes could be


too small to make any statistically significant
decision

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 99


Search Strategy

 Finding an optimal decision tree is NP-hard

 The algorithm presented so far uses a greedy,


top-down, recursive partitioning strategy to
induce a reasonable solution

 Other strategies?
– Bottom-up
– Bi-directional

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 100


Expressiveness

 Decision tree provides expressive representation for


learning discrete-valued function
– But they do not generalize well to certain types of
Boolean functions
 Example: parity function:
– Class = 1 if there is an even number of Boolean attributes with truth
value = True
– Class = 0 if there is an odd number of Boolean attributes with truth
value = True
 For accurate modeling, must have a complete tree

 Not expressive enough for modeling continuous variables


– Particularly when test condition involves only a single
attribute at-a-time

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 101


Decision Boundary
1

0.9

0.8
x < 0.43?

0.7
Yes No
0.6
y

0.5 y < 0.47? y < 0.33?


0.4

0.3
Yes No Yes No

0.2
:4 :0 :0 :4
0.1 :0 :4 :3 :0
0
0 0.1 0.2 0.3 0.4 0.5 0.6 0.7 0.8 0.9 1

x
• Border line between two neighboring regions of different classes is
known as decision boundary
• Decision boundary is parallel to axes because test condition involves
a single attribute at-a-time

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 102


Oblique Decision Trees

x+y<1

Class = + Class =

• Test condition may involve multiple attributes


• More expressive representation
• Finding optimal test condition is computationally expensive
© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 103
Tree Replication
P

Q R

S 0 Q 1

0 1 S 0

0 1

• Same subtree appears in multiple branches

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 104


Model Evaluation

 Metrics for Performance Evaluation


– How to evaluate the performance of a model?

 Methods for Performance Evaluation


– How to obtain reliable estimates?

 Methods for Model Comparison


– How to compare the relative performance
among competing models?

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 105


Model Evaluation

 Metrics for Performance Evaluation


– How to evaluate the performance of a model?

 Methods for Performance Evaluation


– How to obtain reliable estimates?

 Methods for Model Comparison


– How to compare the relative performance
among competing models?

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 106


Metrics for Performance Evaluation

 Focus on the predictive capability of a model


– Rather than how fast it takes to classify or
build models, scalability, etc.
 Confusion Matrix:

PREDICTED CLASS

Class=Yes Class=No
a: TP (true positive)
b: FN (false negative)
Class=Yes a b
ACTUAL c: FP (false positive)
d: TN (true negative)
CLASS Class=No c d

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 107


Metrics for Performance Evaluation…

PREDICTED CLASS

Class=Yes Class=No

Class=Yes a b
ACTUAL (TP) (FN)
CLASS
Class=No c d
(FP) (TN)

 Most widely-used metric:

ad TP  TN
Accuracy  
a  b  c  d TP  TN  FP  FN
© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 108
Limitation of Accuracy

 Consider a 2-class problem


– Number of Class 0 examples = 9990
– Number of Class 1 examples = 10

 If model predicts everything to be class 0,


accuracy is 9990/10000 = 99.9 %
– Accuracy is misleading because model does
not detect any class 1 example

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 109


Cost Matrix

PREDICTED CLASS

C(i|j) Class=Yes Class=No

Class=Yes C(Yes|Yes) C(No|Yes)


ACTUAL
CLASS Class=No C(Yes|No) C(No|No)

C(i|j): Cost of misclassifying class j example as class i

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 110


Computing Cost of Classification

Cost PREDICTED CLASS


Matrix
C(i|j) + -
ACTUAL
+ -1 100
CLASS
- 1 0

Model PREDICTED CLASS Model PREDICTED CLASS


M1 M2
+ - + -
ACTUAL ACTUAL
+ 150 40 + 250 45
CLASS CLASS
- 60 250 - 5 200

Accuracy = 80% Accuracy = 90%


Cost = 3910 Cost = 4255
© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 111
Cost vs Accuracy

Count PREDICTED CLASS Accuracy is proportional to cost if


1. C(Yes|No)=C(No|Yes) = q
Class=Yes Class=No
2. C(Yes|Yes)=C(No|No) = p
Class=Yes a b
ACTUAL N=a+b+c+d
CLASS Class=No c d
Accuracy = (a + d)/N

Cost PREDICTED CLASS


Cost = p (a + d) + q (b + c)
Class=Yes Class=No
= p (a + d) + q (N – a – d)
Class=Yes p q = q N – (q – p)(a + d)
ACTUAL
CLASS Class=No
= N [q – (q-p)  Accuracy]
q p

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 112


Cost-Sensitive Measures
a
Precision (p) 
ac
a
Recall (r) 
ab
2rp 2a
F - measure (F)  
r  p 2a  b  c

 Precision is biased towards C(Yes|Yes) & C(Yes|No)


 Recall is biased towards C(Yes|Yes) & C(No|Yes)
 F-measure is biased towards all except C(No|No)
wa  w d
Weighted Accuracy  1 4

wa  wb wc  w d 1 2 3 4

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 113


Model Evaluation

 Metrics for Performance Evaluation


– How to evaluate the performance of a model?

 Methods for Performance Evaluation


– How to obtain reliable estimates?

 Methods for Model Comparison


– How to compare the relative performance
among competing models?

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 114


Methods for Performance Evaluation

 How to obtain a reliable estimate of


performance?

 Performance of a model may depend on other


factors besides the learning algorithm:
– Class distribution
– Cost of misclassification
– Size of training and test sets

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 115


Learning Curve

 Learning curve shows


how accuracy changes
with varying sample size
 Requires a sampling
schedule for creating
learning curve:
 Arithmetic sampling
(Langley, et al)
 Geometric sampling
(Provost et al)

Effect of small sample size:


- Bias in the estimate
- Variance of estimate

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 116


Methods of Estimation
 Holdout
– Reserve 2/3 for training and 1/3 for testing
 Random subsampling
– Repeated holdout
 Cross validation
– Partition data into k disjoint subsets
– k-fold: train on k-1 partitions, test on the remaining one
– Leave-one-out: k=n
 Stratified sampling
– oversampling vs undersampling
 Bootstrap
– Sampling with replacement

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 117


Model Evaluation

 Metrics for Performance Evaluation


– How to evaluate the performance of a model?

 Methods for Performance Evaluation


– How to obtain reliable estimates?

 Methods for Model Comparison


– How to compare the relative performance
among competing models?

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 118


ROC (Receiver Operating Characteristic)

 Developed in 1950s for signal detection theory to


analyze noisy signals
– Characterize the trade-off between positive
hits and false alarms
 ROC curve plots TP (on the y-axis) against FP
(on the x-axis)
 Performance of each classifier represented as a
point on the ROC curve
– changing the threshold of algorithm, sample
distribution or cost matrix changes the location
of the point

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 119


ROC Curve
- 1-dimensional data set containing 2 classes (positive and negative)
- any points located at x > t is classified as positive

At threshold t:
TP=0.5, FN=0.5, FP=0.12, FN=0.88
© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 120
ROC Curve

(TP,FP):
 (0,0): declare everything
to be negative class
 (1,1): declare everything
to be positive class
 (1,0): ideal

 Diagonal line:
– Random guessing
– Below diagonal line:
 prediction is opposite of
the true class
© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 121
Using ROC for Model Comparison

 No model consistently
outperform the other
 M1 is better for
small FPR
 M2 is better for
large FPR

 Area Under the ROC


curve
 Ideal:
 Area =1
 Random guess:
 Area = 0.5

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 122


How to Construct an ROC curve

• Use classifier that produces


Instance P(+|A) True Class
posterior probability for each
1 0.95 +
test instance P(+|A)
2 0.93 +
3 0.87 - • Sort the instances according
4 0.85 - to P(+|A) in decreasing order
5 0.85 -
• Apply threshold at each
6 0.85 + unique value of P(+|A)
7 0.76 -
8 0.53 + • Count the number of TP, FP,
9 0.43 - TN, FN at each threshold
10 0.25 + • TP rate, TPR = TP/(TP+FN)
• FP rate, FPR = FP/(FP + TN)

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 123


How to construct an ROC curve
Class + - + - - - + - + +
P
Threshold >= 0.25 0.43 0.53 0.76 0.85 0.85 0.85 0.87 0.93 0.95 1.00

TP 5 4 4 3 3 3 3 2 2 1 0

FP 5 5 4 4 3 2 1 1 0 0 0

TN 0 0 1 1 2 3 4 4 5 5 5

FN 0 1 1 2 2 2 2 3 3 4 5

TPR 1 0.8 0.8 0.6 0.6 0.6 0.6 0.4 0.4 0.2 0

FPR 1 1 0.8 0.8 0.6 0.4 0.2 0.2 0 0 0

ROC Curve:

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 124


Test of Significance

 Given two models:


– Model M1: accuracy = 85%, tested on 30 instances
– Model M2: accuracy = 75%, tested on 5000 instances

 Can we say M1 is better than M2?


– How much confidence can we place on accuracy of
M1 and M2?
– Can the difference in performance measure be
explained as a result of random fluctuations in the test
set?

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 125


Confidence Interval for Accuracy

 Prediction can be regarded as a Bernoulli trial


– A Bernoulli trial has 2 possible outcomes
– Possible outcomes for prediction: correct or wrong
– Collection of Bernoulli trials has a Binomial distribution:
 x  Bin(N, p) x: number of correct predictions
 e.g: Toss a fair coin 50 times, how many heads would turn up?
Expected number of heads = Np = 50  0.5 = 25

 Given x (# of correct predictions) or equivalently,


acc=x/N, and N (# of test instances),

Can we predict p (true accuracy of model)?

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 126


Confidence Interval for Accuracy
Area = 1 - 
 For large test sets (N > 30),
– acc has a normal distribution
with mean p and variance
p(1-p)/N

acc  p
P( Z  Z )
p(1  p) / N
 /2 1 / 2

 1 Z/2 Z1-  /2

 Confidence Interval for p:


2  N  acc  Z  Z  4  N  acc  4  N  acc
2 2 2

p  /2  /2

2( N  Z ) 2

 /2

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 127


Confidence Interval for Accuracy

 Consider a model that produces an accuracy of


80% when evaluated on 100 test instances:
– N=100, acc = 0.8 1- Z
– Let 1- = 0.95 (95% confidence)
0.99 2.58
– From probability table, Z/2=1.96
0.98 2.33
N 50 100 500 1000 5000 0.95 1.96

p(lower) 0.670 0.711 0.763 0.774 0.789 0.90 1.65

p(upper) 0.888 0.866 0.833 0.824 0.811

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 128


Comparing Performance of 2 Models

 Given two models, say M1 and M2, which is


better?
– M1 is tested on D1 (size=n1), found error rate = e1
– M2 is tested on D2 (size=n2), found error rate = e2
– Assume D1 and D2 are independent
– If n1 and n2 are sufficiently large, then

e1 ~ N 1 ,  1 
e2 ~ N 2 ,  2 
e (1  e )
– Approximate: ˆ 
i i
i
n i

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 129


Comparing Performance of 2 Models

 To test if performance difference is statistically


significant: d = e1 – e2
– d ~ N(dt,t) where dt is the true difference
– Since D1 and D2 are independent, their variance
adds up:

      ˆ  ˆ
2

t 1
2

2
2 2

1
2

e1(1  e1) e2(1  e2)


 
n1 n2

– At (1-) confidence level, d  d  Z ˆ


t  /2 t

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 130


An Illustrative Example

 Given: M1: n1 = 30, e1 = 0.15


M2: n2 = 5000, e2 = 0.25
 d = |e2 – e1| = 0.1 (2-sided test)

0.15(1  0.15) 0.25(1  0.25)


ˆ 
d
  0.0043
30 5000
 At 95% confidence level, Z/2=1.96

d  0.100  1.96  0.0043  0.100  0.128


t

=> Interval contains 0 => difference may not be


statistically significant
© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 131
Comparing Performance of 2 Algorithms

 Each learning algorithm may produce k models:


– L1 may produce M11 , M12, …, M1k
– L2 may produce M21 , M22, …, M2k
 If models are generated on the same test sets
D1,D2, …, Dk (e.g., via cross-validation)
– For each set: compute dj = e1j – e2j
– dj has mean dt and variance t
– Estimate:  (d  d )
k
2

ˆ  2 j 1 j

k (k  1)
t

d  d  t ˆ
t 1 , k 1 t

© Tan,Steinbach, Kumar Introduction to Data Mining 4/18/2004 132

You might also like