IF YOU WOULD LIKE TO GET AN ACCOUNT, please write an
email to s dot adaszewski at gmail dot com. User accounts
are meant only to report issues and/or generate pull requests.
This is a purpose-specific Git hosting for
ADARED
projects. Thank you for your understanding!
Stanislaw Adaszewski
5 years ago
1 changed files with 4 additions and 1 deletions
Unified View
Diff Options
+ 4
- 1
tests/triacontagon/test_sampling.py
View File
| @@ -20,16 +20,19 @@ def test_get_true_classes_01(): | |||||
| print('true_classes:', true_classes) | print('true_classes:', true_classes) | ||||
| assert torch.all(true_classes == torch.tensor([ | assert torch.all(true_classes == torch.tensor([ | ||||
| [1, 3], | |||||
| [1, 3], | [1, 3], | ||||
| [4, -1], | [4, -1], | ||||
| [0, 1], | [0, 1], | ||||
| [0, 1], | |||||
| [2, 4], | |||||
| [2, 4], | [2, 4], | ||||
| [1, -1] | [1, -1] | ||||
| ])) | ])) | ||||
| def test_get_true_classes_02(): | def test_get_true_classes_02(): | ||||
| adj_mat = torch.rand(2000, 2000).round().to_sparse() | |||||
| adj_mat = (torch.rand(2000, 2000) < 0.1).to_sparse() | |||||
| t = time.time() | t = time.time() | ||||
| true_classes = get_true_classes(adj_mat) | true_classes = get_true_classes(adj_mat) | ||||