This is a Test

[1]:

from jlpy.pytorch.base.linear_regression import base_impl as bi

b = bi.BaseImpl(bsize=32, nepoch=5, gap=50, ws=[2, -3.4, 5, -1, 9, 23, -16.12, 22, 33, 55.55, -19.22, 88.23], b=4.2, n=0.01, wd=0.2)
b.fit()
b.plot()

/home/pne/env/jls/lib/python3.11/site-packages/torch/nn/modules/lazy.py:181: UserWarning: Lazy modules are a new feature under heavy development so changes to the API or functionality can happen at any moment.
  warnings.warn('Lazy modules are a new feature under heavy development '

[2]:

from jlpy.pytorch.base.attention_mechanism.base_impl import AMBase
import torch


amb = AMBase()
ts = torch.eye(10)
amb.show_heatmap(ts)