turnaround5954
/ ple.py
Created
February 28, 2022 04:51
A naive implementation of Progressive Layered Extraction (PLE) in pytorch
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| """ | |
| Progressive Layered Extraction. | |
| """ | |
| import torch | |
| import torch.nn as nn | |
| class CGCLayer(nn.Module): | |
| """ |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| """ | |
| MIND: Multi-Interest Network with Dynamic Routing. | |
| """ | |
| import torch | |
| import torch.nn as nn | |
| import numpy as np | |
| import mind_capsule | |
| import sampled_softmax |
turnaround5954
/ mind_capsule.py
Last active
February 28, 2022 04:34
A naive implementation of capsule layer in MIND model
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| """ | |
| Capsule layer (MIND) in pytorch. | |
| """ | |
| import torch | |
| import torch.nn as nn | |
| torch.manual_seed(0) | |
| class MINDCapsule(nn.Module): |
turnaround5954
/ sampled_softmax.py
Last active
February 28, 2022 04:34
A naive implementation of sampled softmax (pytorch)
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| """ | |
| Sampled softmax in pytorch. | |
| """ | |
| import torch | |
| import torch.nn as nn | |
| import numpy as np | |
| torch.manual_seed(0) | |