HyperTab: hypernetwork for small tabular datasets
Project description
HyperTab
HyperTab is a hypernetwork-based classifier for small tabular datasets.
Installation
pip install hypertab
Usage
from hypertab import HyperTabClassifier
DEVICE = "cuda:0" if torch.cuda.is_available() else "cpu"
clf = HyperTabClassifier(0.2, device=DEVICE, test_nodes=100, epochs=10, hidden_dims=5)
clf.fit(X, y)
clf.predict(X)
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
hypertab-0.1.5.tar.gz
(9.2 kB
view hashes)