CODEWITHSUNDEEP
pythonmachine-learningscikit-learn
OM MEHTA
OM MEHTA

Reputation: 11

Custom data prediction using decision trees

I am using train_test_split to train the model and check the results using predict. How do I proceed to predict the labels of additional data, for example, from a test set or from user inputs?

Upvotes: 0

Views: 105

Answers (1)

secretive
secretive

Reputation: 2104

You can use the custom data for prediction as long as it has the same number, order and type of features as your training data into an array type, not in list. If you meet these conditions, you can send that array to the model for prediction with the normal predict() method.

Upvotes: 1

Related Questions