SKLearner Home | About | Contact | Examples

Scikit-Learn PassiveAggressiveClassifier Model

Passive-Aggressive Classification is an efficient algorithm for handling large-scale text classification tasks with real-time updates. It is particularly suited for online learning scenarios where the model updates itself with each new data instance.

The key hyperparameters of PassiveAggressiveClassifier include C (regularization parameter), which controls the trade-off between achieving a low training error and a low testing error, and max_iter (maximum number of iterations), which sets the limit for the number of passes over the training data.

This algorithm is appropriate for binary classification tasks, especially where the data is received in a sequential or streaming manner.

from sklearn.datasets import make_classification
from sklearn.model_selection import train_test_split
from sklearn.linear_model import PassiveAggressiveClassifier
from sklearn.metrics import accuracy_score

# generate binary classification dataset
X, y = make_classification(n_samples=100, n_features=5, n_classes=2, random_state=1)

# split into train and test sets
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=1)

# create model
model = PassiveAggressiveClassifier(max_iter=1000, random_state=1)

# fit model
model.fit(X_train, y_train)

# evaluate model
yhat = model.predict(X_test)
acc = accuracy_score(y_test, yhat)
print('Accuracy: %.3f' % acc)

# make a prediction
row = [[-1.10325445, -0.49821356, -0.05962247, -0.89224592, -0.70158632]]
yhat = model.predict(row)
print('Predicted: %d' % yhat[0])

Running the example gives an output like:

Accuracy: 0.800
Predicted: 0

The steps are as follows:

  1. First, a synthetic binary classification dataset is generated using the make_classification() function. This creates a dataset with a specified number of samples (n_samples), features (n_features), classes (n_classes), and a fixed random seed (random_state) for reproducibility. The dataset is split into training and test sets using train_test_split().

  2. Next, a PassiveAggressiveClassifier model is instantiated with max_iter set to 1000 and a fixed random seed for reproducibility. The model is then fit on the training data using the fit() method.

  3. The performance of the model is evaluated by comparing the predictions (yhat) to the actual values (y_test) using the accuracy score metric.

  4. A single prediction can be made by passing a new data sample to the predict() method.

This example demonstrates how to quickly set up and use a PassiveAggressiveClassifier for binary classification tasks, showcasing the efficiency and suitability of this algorithm for online learning scenarios in scikit-learn. The model can be fit directly on the training data and used to make predictions on new data, enabling its use in real-world classification problems with streaming or sequentially arriving data.



See Also