Skip to content

Conversation

@chijames
Copy link
Contributor

@chijames chijames commented Jan 6, 2021

Hi,

I implement the feature as stated in title. The implementation basically follows section 3.3 of [1] with similar gradient update rules for marginals (like those in 3.2). I check marginals using the log-partition/marginals identity and they are the same. Let me know if anything needs to be improved!

[1] Structured Prediction Models via the Matrix-Tree Theorem. Koo et al. 2007.

@srush
Copy link
Collaborator

srush commented Jan 11, 2021

This is great thanks for contributing.

Sorry the repo is having some issues due to the travis. I will fix it up this week and add tests.

chijames and others added 2 commits January 14, 2021 05:35
@srush srush merged commit 9083dd4 into harvardnlp:master Jan 15, 2021
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants