scikit-learn estimators
Estimators that allows for the standardized implementation and testing of algorithms a common, lightweight interface for classifiers to follow. By using this interface, we can apply these tools to arbitrary classifiers, without needing to worry about how the algorithms work.
Estimators must have the following two important functions:
- fit(): This function performs the training of the algorithm - setting the values of internal parameters. The fit() takes two inputs, the training sample dataset and the corresponding classes for those samples.
- predict(): This the class of the testing samples that we provide as the only input. This function returns a NumPy array with the predictions of each input testing sample.
Most scikit-learn estimators use NumPy arrays or a related format for input and output. However this is by convention and not required to use the interface.
There are many estimators implemented in scikit-learn and more in other open source projects that use the same interface. These (SVM), random forests. We will use many
of these algorithms in later chapters. In this chapter, we will use the nearest neighbor
algorithm.
For this chapter, you will need to install a new library called matplotlib. The easiest way to install it is to use pip3, as you did in Chapter 1, Getting Started with Data Mining, to install scikit-learn:
$pip3 install matplotlib
If you have matplotlib, seek the official installation instructions at:
http://matplotlib.org/users/installing.html