Scattering Compositional Learner for solving Raven's Progressive Matrices
Project description
Scattering Compositional Learner
PyTorch implementation of Scattering Compositional Learner [1] for solving Raven's Progressive Matrices.
Setup
$ pip install scattering_compositional_learner
Usage
import torch
from scattering_compositional_learner import ScatteringCompositionalLearner
x = torch.rand(4, 16, 160, 160)
scl = ScatteringCompositionalLearner(image_size=160)
logits = scl(x)
y_hat = logits.log_softmax(dim=-1)
y_hat # torch.Tensor with shape (4, 8)
Unit tests
$ python -m pytest tests
Alternative implementations
The same model was additionally implemented by:
Bibliography
[1] Wu, Yuhuai, et al. "The Scattering Compositional Learner: Discovering Objects, Attributes, Relationships in Analogical Reasoning." arXiv preprint arXiv:2007.04212 (2020).
Citations
@article{wu2020scattering,
title={The Scattering Compositional Learner: Discovering Objects, Attributes, Relationships in Analogical Reasoning},
author={Wu, Yuhuai and Dong, Honghua and Grosse, Roger and Ba, Jimmy},
journal={arXiv preprint arXiv:2007.04212},
year={2020}
}
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Close
Hashes for scattering_compositional_learner-0.1.0.tar.gz
Algorithm | Hash digest | |
---|---|---|
SHA256 | 76591875c09dccf01ee5f06a841b350357f6f5678da616ccf2a55e3ee76dc7cd |
|
MD5 | 3a809234b41ec9a5cf8e495f824d369b |
|
BLAKE2b-256 | 90d4fbfa96dd29e24eabaebb1024a9b707cf333aa2516193807e75dbce69a7cc |
Close
Hashes for scattering_compositional_learner-0.1.0-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 6d3c3d17f13b3ddb56b9a542552a4c02f87c9e4202a6993b07cf5d777377fa14 |
|
MD5 | 7ebf66bb2bd9db3c9263e0f1e0983e6a |
|
BLAKE2b-256 | 496b3f309d8a25d6f937f27bd28773bf6794fb6ac0053bd8da29ecf9b6bc5160 |