Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
da03 committed Sep 3, 2019
1 parent 165be00 commit 48cc96e
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -30,15 +30,15 @@ scores = torch.rand(N, 100, 100, requires_grad=True)
marginals = struct.deptree(scores)

# Tree Argmax
argmax = struct.deptree(scores, seminring=struct.MaxSemiring)
argmax = struct.deptree(scores, semiring=struct.MaxSemiring)
max_score = torch.mul(argmax, scores)

# Tree Counts
ones = torch.ones(N, 100, 100)
ntrees = struct.deptree(ones, semiring=struct.StdSemiring)

# Tree Sample
sample = struct.deptree(scores, seminring=struct.SampledSemiring)
sample = struct.deptree(scores, semiring=struct.SampledSemiring)

# Tree Partition
v, _ = struct.deptree_inside(scores)
Expand Down

0 comments on commit 48cc96e

Please sign in to comment.