Skip to main content
Revision 363d4612debf181596ea4dce956ddaa60cd49828 authored by Rafael Ballester-Ripoll on 16 September 2022, 15:04:27 UTC, committed by GitHub on 16 September 2022, 15:04:27 UTC
Merge pull request #39 from Aelphy/main
Fixes #38
2 parent s 87ba967 + 37ff87b
Raw File
test_interpolation.py
import numpy as np
import tntorch as tn
import torch
torch.set_default_dtype(torch.float64)


def test_als_completion():
    I = 8
    train_x = torch.arange(I)[:, None].repeat(1, 2)
    train_y = torch.ones(I)
    t = tn.als_completion(train_x, train_y, ranks_tt=3)
    assert tn.relative_error(train_y, t[train_x]) < 1e-5
back to top