Skip to content

sylwekczmil/sevq

Repository files navigation

sevq

Documentation Status

SEVQ: Simple Evolving Vector Quantization

Installation

To install sevq, run this command in your terminal:

pip install sevq

Usage

Training and prediction one sample at a time

from sevq.algorithm import SEVQ

c = SEVQ()
c.partial_fit([-2, -2], 2)
c.partial_fit([-1, -1], 1)
c.partial_fit([1, 1], 1)
c.partial_fit([2, 2], 2)

print(c.predict([0, 0]))  # 1
print(c.predict([3, 3]))  # 2
print(c.predict([-3, -3]))  # 2

Training and prediction on multiple samples

from sevq.algorithm import SEVQ

c = SEVQ()
c.fit(
    [[-2, -2], [-1, -1], [1, 1], [2, 2]],
    [2, 1, 1, 2],
    epochs=1, permute=False
)

print(c.predict([[0, 0], [3, 3], [-3, -3]]))  # [1, 2, 2]