forked from aimacode/aima-python
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathtest_learning.py
More file actions
141 lines (93 loc) · 3.82 KB
/
Copy pathtest_learning.py
File metadata and controls
141 lines (93 loc) · 3.82 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
from learning import parse_csv, weighted_mode, weighted_replicate, DataSet, \
PluralityLearner, NaiveBayesLearner, NearestNeighborLearner, \
NeuralNetLearner, PerceptronLearner, DecisionTreeLearner, \
euclidean_distance, grade_learner, err_ratio, random_weights
from utils import DataFile
def test_euclidean():
distance = euclidean_distance([1, 2], [3, 4])
assert round(distance, 2) == 2.83
distance = euclidean_distance([1, 2, 3], [4, 5, 6])
assert round(distance, 2) == 5.2
distance = euclidean_distance([0, 0, 0], [0, 0, 0])
assert distance == 0
def test_exclude():
iris = DataSet(name='iris', exclude=[3])
assert iris.inputs == [0, 1, 2]
def test_parse_csv():
Iris = DataFile('iris.csv').read()
assert parse_csv(Iris)[0] == [5.1, 3.5, 1.4, 0.2,'setosa']
def test_weighted_mode():
assert weighted_mode('abbaa', [1, 2, 3, 1, 2]) == 'b'
def test_weighted_replicate():
assert weighted_replicate('ABC', [1, 2, 1], 4) == ['A', 'B', 'B', 'C']
def test_means_and_deviation():
iris = DataSet(name="iris")
means, deviations = iris.find_means_and_deviations()
assert round(means["setosa"][0], 3) == 5.006
assert round(means["versicolor"][0], 3) == 5.936
assert round(means["virginica"][0], 3) == 6.588
assert round(deviations["setosa"][0], 3) == 0.352
assert round(deviations["versicolor"][0], 3) == 0.516
assert round(deviations["virginica"][0], 3) == 0.636
def test_plurality_learner():
zoo = DataSet(name="zoo")
pL = PluralityLearner(zoo)
assert pL([1, 0, 0, 1, 0, 0, 0, 1, 1, 1, 0, 0, 4, 1, 0, 1]) == "mammal"
def test_naive_bayes():
iris = DataSet(name="iris")
# Discrete
nBD = NaiveBayesLearner(iris, continuous=False)
assert nBD([5, 3, 1, 0.1]) == "setosa"
assert nBD([6, 3, 4, 1.1]) == "versicolor"
assert nBD([7.7, 3, 6, 2]) == "virginica"
# Continuous
nBC = NaiveBayesLearner(iris, continuous=True)
assert nBC([5, 3, 1, 0.1]) == "setosa"
assert nBC([6, 5, 3, 1.5]) == "versicolor"
assert nBC([7, 3, 6.5, 2]) == "virginica"
def test_k_nearest_neighbors():
iris = DataSet(name="iris")
kNN = NearestNeighborLearner(iris,k=3)
assert kNN([5, 3, 1, 0.1]) == "setosa"
assert kNN([6, 5, 3, 1.5]) == "versicolor"
assert kNN([7.5, 4, 6, 2]) == "virginica"
def test_decision_tree_learner():
iris = DataSet(name="iris")
dTL = DecisionTreeLearner(iris)
assert dTL([5, 3, 1, 0.1]) == "setosa"
assert dTL([6, 5, 3, 1.5]) == "versicolor"
assert dTL([7.5, 4, 6, 2]) == "virginica"
def test_neural_network_learner():
iris = DataSet(name="iris")
classes = ["setosa","versicolor","virginica"]
iris.classes_to_numbers(classes)
nNL = NeuralNetLearner(iris, [5], 0.15, 75)
tests = [([5, 3, 1, 0.1], 0),
([5, 3.5, 1, 0], 0),
([6, 3, 4, 1.1], 1),
([6, 2, 3.5, 1], 1),
([7.5, 4, 6, 2], 2),
([7, 3, 6, 2.5], 2)]
assert grade_learner(nNL, tests) >= 2/3
assert err_ratio(nNL, iris) < 0.25
def test_perceptron():
iris = DataSet(name="iris")
iris.classes_to_numbers()
classes_number = len(iris.values[iris.target])
perceptron = PerceptronLearner(iris)
tests = [([5, 3, 1, 0.1], 0),
([5, 3.5, 1, 0], 0),
([6, 3, 4, 1.1], 1),
([6, 2, 3.5, 1], 1),
([7.5, 4, 6, 2], 2),
([7, 3, 6, 2.5], 2)]
assert grade_learner(perceptron, tests) > 1/2
assert err_ratio(perceptron, iris) < 0.4
def test_random_weights():
min_value = -0.5
max_value = 0.5
num_weights = 10
test_weights = random_weights(min_value, max_value, num_weights)
assert len(test_weights) == num_weights
for weight in test_weights:
assert weight >= min_value and weight <= max_value