Ross Johnstone

OverflowError: n_iter too high in bpr.FMRecommender

1

When I try to set a massive number of iterations, e.g. ```python fm = bpr.FMRecommender(n_iter=10000000000000, init_stdev=0.1, step_size=0.01, rank=10, random_state=123) fm.fit(X, compares) ``` I get as an error (which occurs when...

Error when tuning NGBSurvival with GridSearchCV

1

Minimal example: ``` from ngboost import NGBSurvival import numpy.random as npr from sklearn.model_selection import GridSearchCV X = npr.randn(100, 5) T = npr.rand(100) E = npr.randint(2, size=100) param_grid = {"learning_rate": [0.01,...

Question: what datasets were pre-trained models pre-trained on?

1

Some of the pre-trained models are just described as "pre-trained", while others are described as "pre-trained then fine-tuned on _x_". What data was the original pre-trained performed on, and for...

Addresses #9631. Implements `LinkPredMRR` as a `LinkPredMetric`, with test included.

### 🚀 The feature, motivation and pitch [MRR](https://en.wikipedia.org/wiki/Mean_reciprocal_rank) is another common metric for recommendations. There are a few uses of it throughout the codebase, but it would be good to...