Skip to content

Add Stochastic Multiplicative Updates#23

Open
bantin wants to merge 7 commits intomasterfrom
stochastic
Open

Add Stochastic Multiplicative Updates#23
bantin wants to merge 7 commits intomasterfrom
stochastic

Conversation

@bantin
Copy link
Collaborator

@bantin bantin commented Nov 21, 2018

No description provided.

Copy link
Collaborator

@ahwillia ahwillia left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Nice work! Does this enable us to fit to the full Maze dataset?

Small notes:

  • I wonder if it makes sense to define a separate StochasticMU class.
  • What is the intended API? Something like model = CMF(alg_name="stochastic_mu")?

Thanks again - excited to try this out!

@bantin
Copy link
Collaborator Author

bantin commented Nov 21, 2018 via email

@bantin
Copy link
Collaborator Author

bantin commented Nov 21, 2018 via email

@ahwillia
Copy link
Collaborator

see some suggestions in #24

Proposed changes to stochastic mult update algorithm. Creates a separate StochasticMultUpdate class.
The cache_resids function now only updates the section of the residuals
matrix which corresponds to current window of the data matrix.
@bantin
Copy link
Collaborator Author

bantin commented Nov 22, 2018

@ahwillia Updated the cache_resids function as you suggested. Thoughts?

@ahwillia
Copy link
Collaborator

Before merging, can you add a small notebook testing this out on the Maze dataset?

@bantin
Copy link
Collaborator Author

bantin commented Nov 22, 2018 via email

@degleris1
Copy link
Owner

degleris1 commented Nov 23, 2018

Merge issues (just in __init__.py) resolved in the commit above. The stochastic update rule looks good to me!

@bantin
Copy link
Collaborator Author

bantin commented Dec 12, 2018

@ahwillia Just added a simple notebook that shows fitting a matrix with 20k timebins in 8.5 mins. Admittedly, the fit isn't great, but if you think it's good enough to show the stochastic functionality, we can probably merge this to master.

Copy link
Owner

@degleris1 degleris1 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Looks good!

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.

3 participants