From 2ece052a889be35d8b067af2549b6acbe2db6f1d Mon Sep 17 00:00:00 2001 From: alxnkt Date: Wed, 5 Jan 2022 12:09:57 +0300 Subject: [PATCH] Correct income_classifier.py Fix issue about using LabelEncoder.transform and np.array functions, as they accept only arrays. --- Chapter 02/code/income_classifier.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/Chapter 02/code/income_classifier.py b/Chapter 02/code/income_classifier.py index 740095f..3425eb2 100644 --- a/Chapter 02/code/income_classifier.py +++ b/Chapter 02/code/income_classifier.py @@ -75,10 +75,10 @@ if item.isdigit(): input_data_encoded[i] = int(input_data[i]) else: - input_data_encoded[i] = int(label_encoder[count].transform(input_data[i])) + input_data_encoded[i] = int(label_encoder[count].transform([input_data[i]])) count += 1 -input_data_encoded = np.array(input_data_encoded) +input_data_encoded = np.array([input_data_encoded]) # Run classifier on encoded datapoint and print output predicted_class = classifier.predict(input_data_encoded)