def testPredictFivePoints(self):
medoids = [4, 12, 25, 37]
to_predict = [[0.3, 0.2], [4.1, 1.1], [3.9, 1.1], [2.1, 1.9],
[2.1, 4.1]]
kmedoids_test_template.templatePredict(SIMPLE_SAMPLES.SAMPLE_SIMPLE3,
medoids, to_predict,
[0, 3, 3, 2, 1], False)
def testPredictOnePointUserMetric(self):
medoids = [4, 12, 25, 37]
metric = distance_metric(type_metric.USER_DEFINED,
func=distance_metric(type_metric.EUCLIDEAN))
kmedoids_test_template.templatePredict(SIMPLE_SAMPLES.SAMPLE_SIMPLE3,
medoids, [[0.3, 0.2]], [0],
False,
metric=metric)
def testPredictTwoPoints(self):
medoids = [4, 12, 25, 37]
kmedoids_test_template.templatePredict(SIMPLE_SAMPLES.SAMPLE_SIMPLE3,
medoids,
[[0.3, 0.2], [2.1, 1.9]],
[0, 2], False)
kmedoids_test_template.templatePredict(SIMPLE_SAMPLES.SAMPLE_SIMPLE3,
medoids,
[[2.1, 4.1], [2.1, 1.9]],
[1, 2], False)
def testPredictFivePointsUserMetric(self):
medoids = [4, 12, 25, 37]
to_predict = [[0.3, 0.2], [4.1, 1.1], [3.9, 1.1], [2.1, 1.9],
[2.1, 4.1]]
metric = distance_metric(type_metric.USER_DEFINED,
func=distance_metric(type_metric.EUCLIDEAN))
kmedoids_test_template.templatePredict(SIMPLE_SAMPLES.SAMPLE_SIMPLE3,
medoids,
to_predict, [0, 3, 3, 2, 1],
False,
metric=metric)
def testPredictOnePoint(self):
medoids = [4, 12, 25, 37]
kmedoids_test_template.templatePredict(SIMPLE_SAMPLES.SAMPLE_SIMPLE3,
medoids, [[0.3, 0.2]], [0],
False)
kmedoids_test_template.templatePredict(SIMPLE_SAMPLES.SAMPLE_SIMPLE3,
medoids, [[4.1, 1.1]], [3],
False)
kmedoids_test_template.templatePredict(SIMPLE_SAMPLES.SAMPLE_SIMPLE3,
medoids, [[2.1, 1.9]], [2],
False)
kmedoids_test_template.templatePredict(SIMPLE_SAMPLES.SAMPLE_SIMPLE3,
medoids, [[2.1, 4.1]], [1],
False)
