OBAN Classifier: A Skorch-based flexible neural network for binary and multiclass classification
Project description
OBAN Classifier
Oban Classifier is a flexible neural network-based classifier built on top of PyTorch and Skorch. It supports both binary and multiclass classification, and allows users to define parameters such as the number of units, activation function, dropout rate, and more.
Features
- Supports binary and multiclass classification.
- Allows user-defined parameters for hidden units, activation functions, dropout, and more.
- Built using Skorch and PyTorch for easy integration with scikit-learn pipelines.
- Provides detailed performance metrics including accuracy, precision, recall, F1-score, and confusion matrix.
Installation
You can install the library via pip after publishing it on PyPI:
pip install oban_classifier
### Usage Example
```python
from oban_classifier import oban_classifier, post_classification_analysis, plot_lime_importance
from sklearn.datasets import load_breast_cancer
from sklearn.preprocessing import StandardScaler
import pandas as pd
# Load the Breast Cancer dataset
data = load_breast_cancer()
X = pd.DataFrame(data.data, columns=data.feature_names)
y = pd.Series(data.target)
# Train and evaluate the model (num_classes is automatically inferred)
netv, X_test, y_test = oban_classifier(X, y, num_units=128, max_epochs=80, lr=0.001)
# Convert X_test to DataFrame with feature names
X_test_df = pd.DataFrame(X_test, columns=X.columns)
# Predict probabilities
y_proba = netv.predict_proba(X_test_df.to_numpy())
# Perform post-classification analysis
post_classification_analysis(X_test_df, y_test, y_proba, threshold=0.5)
# Explain predictions using LIME with correct feature names
plot_lime_importance(netv, X_test_df, y_test, feature_names=X.columns)
# Predict on new data
new_data = pd.DataFrame([[15.0, 20.0, 85.0, 60.0, 0.5, 1.5, 3.0, 0.02, 0.2, 0.3,
0.1, 25.0, 50.0, 150.0, 100.0, 0.1, 0.5, 2.5, 0.01, 0.1,
15.0, 20.0, 85.0, 60.0, 0.5, 1.5, 3.0, 0.02, 0.2, 0.3]],
columns=X.columns)
# Normalize the new data using the same scaler as before
scaler = StandardScaler()
scaler.fit(X)
new_data_scaled = scaler.transform(new_data)
# Predict the class for the new data
predicted_class = netv.predict(new_data_scaled)
print(f"Predicted class: {predicted_class}")
# Predict probabilities for the new data
predicted_probabilities = netv.predict_proba(new_data_scaled)
print(f"Predicted probabilities: {predicted_probabilities}")
#### oban_classifier Parameters
X (pd.DataFrame): The feature matrix. Should be a Pandas DataFrame where each row is an instance and each column is a feature.
y (pd.Series): The target variable. Should be a Pandas Series where each value corresponds to the target class of a given row in X.
num_units (int, optional, default=128): The number of hidden units in the dense layers of the neural network.
nonlin (torch.nn.Module, optional, default=nn.ReLU()): The non-linear activation function to apply after each dense layer. Default is ReLU, but can be changed to other functions like nn.Sigmoid() or nn.Tanh().
dropout_rate (float, optional, default=0.5): The dropout rate applied to the layers to prevent overfitting. Should be between 0 and 1.
max_epochs (int, optional, default=10): The maximum number of epochs to train the model.
lr (float, optional, default=0.01): The learning rate for the optimizer.
test_size (float, optional, default=0.2): The proportion of the dataset to be used for testing. Should be between 0 and 1.
random_state (int, optional, default=42): The seed for the random number generator to ensure reproducible results during dataset splitting.
#### post_classification_analysis Parameters
X (pd.DataFrame): The feature matrix used during testing.
y_true (pd.Series): The true class labels for the test set.
y_proba (np.ndarray): The predicted probabilities for each class.
threshold (float, optional, default=0.5): The decision threshold used for binary classification. Predictions with probabilities greater than or equal to the threshold are classified as 1, otherwise as 0. This parameter is ignored in multiclass classification.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
File details
Details for the file oban_classifier-0.1.19.11.tar.gz
.
File metadata
- Download URL: oban_classifier-0.1.19.11.tar.gz
- Upload date:
- Size: 5.7 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/5.1.1 CPython/3.9.19
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | 8f65bce09caa37b53af9887231f084fa42125719b8de001bbb54f0eba275332e |
|
MD5 | 895cfac5d9af099473fb6b188e14a46d |
|
BLAKE2b-256 | 166f402b3988921e24a63626b619a3d24f6b271ddefd46c5d758991c8f2ef135 |
File details
Details for the file oban_classifier-0.1.19.11-py3-none-any.whl
.
File metadata
- Download URL: oban_classifier-0.1.19.11-py3-none-any.whl
- Upload date:
- Size: 6.2 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/5.1.1 CPython/3.9.19
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | a096f8a80139e371b70a656ea9bae5ea58cd1606f9ac107f62fa7a8b1843b509 |
|
MD5 | 3f8bc0b851d1b6f64ae8f7b75ed68aaa |
|
BLAKE2b-256 | 2a1d87c7bf6138a07e3ea793d9b64a9fbad86134b0466ee6aace5cd98ca2b9d9 |