Training the baseline model

In this section, we will perform actual training using the following ML algorithms. This step is time-consuming as it needs more computation power. We use 75% of the training dataset for actual training and 25% of the dataset for testing in order to measure the training accuracy.

You can find the code snippet in the following figure:

Training the baseline model

Figure 1.52: Code snippet for performing training

In the preceding code snippet, you can see that we performed the actual training operation using the fit() function from the scikit-learn library. This function uses the given parameter and trains the model by taking the input of the target data ...

Get Machine Learning Solutions now with the O’Reilly learning platform.

O’Reilly members experience books, live events, courses curated by job role, and more from O’Reilly and nearly 200 top publishers.