def _create_model_from_dialog(cls, dialog): # create prior alpha from user choice # prior strength multiplies the dirichlet parameters alpha alpha = (np.array(ALPHA_DEFAULT) - 1.) * dialog.prior_strength + 1. alpha = create_band_matrix(dialog.nclasses, alpha) model = ModelB.create_initial_state(dialog.nclasses, dialog.nannotators, alpha=alpha) return model
def default_alpha(nclasses): return create_band_matrix(nclasses, ALPHA_DEFAULT)