def test_recommend_num_components(self): dataset = self.create_otago_dataset() # eigenvalues = [0.0490834 1.28402771] # 96.3% of variance in first principal component recommendation1 = pca.recommend_num_components(dataset, 0.95) self.assertEqual(recommendation1, 1) recommendation2 = pca.recommend_num_components(dataset, 0.97) self.assertEqual(recommendation2, 2)
def test_default_recommended_num_components(self): dataset = self.create_otago_dataset() self.assertEqual(pca.recommend_num_components(dataset), 1)