Skip to content

[MRG] Added sklearn wrapper for AuthorTopic model#1403

Merged
menshikh-iv merged 16 commits into
piskvorky:developfrom
chinmayapancholi13:atm_sklearn_wrapper
Jun 28, 2017
Merged

[MRG] Added sklearn wrapper for AuthorTopic model#1403
menshikh-iv merged 16 commits into
piskvorky:developfrom
chinmayapancholi13:atm_sklearn_wrapper

Conversation

@chinmayapancholi13
Copy link
Copy Markdown
Contributor

This PR adds scikit-learn wrapper for Gensim's AuthorTopic model.

@menshikh-iv
Copy link
Copy Markdown
Contributor

As well as comments in PR 1405

@chinmayapancholi13 chinmayapancholi13 changed the title [WIP] Added sklearn wrapper for AuthorTopic model [MRG] Added sklearn wrapper for AuthorTopic model Jun 28, 2017
@menshikh-iv menshikh-iv merged commit 7ccaabc into piskvorky:develop Jun 28, 2017
saparina pushed a commit to saparina/gensim that referenced this pull request Jul 9, 2017
* added sklearn wrapper for author topic model

* added 'fit', 'partial_fit', 'transform' functions

* added unit-tests for ATModel

* refactored code acc. to composite design pattern

* refactored wrapper and tests

* removed 'self.corpus' attribute

* updates 'self.model' to 'self.gensim_model'

* updated 'fit' and 'transform' functions

* updated 'testTransform' test

* updated 'tranform' function slightly

* updated 'testTransform' test

* PEP8 change

* updated 'testTransform' test

* included 'NotFittedError' error

* added pipeline unittest for ATModel
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