cair / tmu

Implements the Tsetlin Machine, Coalesced Tsetlin Machine, Convolutional Tsetlin Machine, Regression Tsetlin Machine, and Weighted Tsetlin Machine, with support for continuous features, drop clause, Type III Feedback, focused negative sampling, multi-task classifier, autoencoder, literal budget, and one-vs-one multi-class classifier. TMU is written in Python with wrappers for C and CUDA-based clause evaluation and updating.
https://pypi.org/project/tmu/
MIT License
130 stars 14 forks source link

Feature refactor models #5

Closed perara closed 1 year ago

perara commented 1 year ago

The following pull request refactors the tsetlin_machine.py, splitting it into separate files, located in respective categories of classification, regression, and autoencoder.

I've also added the same structure for the examples.

Does this looks OK @olegranmo?