@Test public void testNoUsers() { BiasModel model = new UserBiasModel(1.5, Long2DoubleMaps.EMPTY_MAP); assertThat(model.getIntercept(), equalTo(1.5)); assertThat(model.getItemBias(42L), equalTo(0.0)); assertThat(model.getUserBias(42L), equalTo(0.0)); }
@Test public void testWithUsers() { BiasModel model = new UserBiasModel(1.5, Long2DoubleMaps.singleton(42L, 1.0)); assertThat(model.getIntercept(), equalTo(1.5)); assertThat(model.getUserBias(42L), equalTo(1.0)); assertThat(model.getUserBias(37L), equalTo(0.0)); assertThat(model.getItemBias(42L), equalTo(0.0)); }
return new UserBiasModel(mean, offsets);
return new UserBiasModel(mean, offsets);