skorch-dev/skorch

This is stars and forks stats for /skorch-dev/skorch repository. As of 27 Apr, 2024 this repository has 5348 stars and 363 forks.

A scikit-learn compatible neural network library that wraps PyTorch. Resources Documentation Source Code Installation Examples To see more elaborate examples, look here. import numpy as np from sklearn.datasets import make_classification from torch import nn from skorch import NeuralNetClassifier X, y = make_classification(1000, 20, n_informative=10, random_state=0) X = X.astype(np.float32) y = y.astype(np.int64) class MyModule(nn.Module): def __init__(self, num_units=10, nonlin=nn.ReLU()): ...