def test_scalar_prediction_type(): model = Workspace(quiet=True) model.learn("1 | a b c") assert model.get_prediction_type() == model.pSCALAR prediction = model.predict(" | a b c") assert isinstance(prediction, float) del model
def test_multilabel_prediction_type(): model = Workspace(multilabel_oaa=4, quiet=True) model.learn("1 | a b c") assert model.get_prediction_type() == model.pMULTILABELS prediction = model.predict(" | a b c") assert isinstance(prediction, list) del model
def test_action_probs_prediction_type(): model = Workspace(cb_explore=2, ngram=2, quiet=True) model.learn("1 | a b c") assert model.get_prediction_type() == model.pACTION_PROBS prediction = model.predict(" | a b c") assert isinstance(prediction, list) del model
def test_multiclass_prediction_type(): n = 3 model = Workspace(loss_function="logistic", oaa=n, quiet=True) model.learn("1 | a b c") assert model.get_prediction_type( ) == vowpalwabbit.PredictionType.MULTICLASS prediction = model.predict(" | a b c") assert isinstance(prediction, int) del model
def test_action_scores_prediction_type(): model = Workspace(loss_function="logistic", csoaa_ldf="m", quiet=True) multi_ex = [model.example("1:1 | a b c"), model.example("2:-1 | a b c")] model.learn(multi_ex) assert model.get_prediction_type() == model.pMULTICLASS multi_ex = [model.example("1 | a b c"), model.example("2 | a b c")] prediction = model.predict(multi_ex) assert isinstance(prediction, int) del model
def test_scalars_prediction_type(): n = 3 model = Workspace(loss_function="logistic", oaa=n, probabilities=True, quiet=True) model.learn("1 | a b c") assert model.get_prediction_type() == vowpalwabbit.PredictionType.SCALARS prediction = model.predict(" | a b c") assert isinstance(prediction, list) assert len(prediction) == n del model
def test_prob_prediction_type(): model = Workspace( loss_function="logistic", csoaa_ldf="mc", probabilities=True, quiet=True, ) multi_ex = [ model.example("1:0.2 | a b c"), model.example("2:0.8 | a b c"), ] model.learn(multi_ex) assert model.get_prediction_type() == vowpalwabbit.PredictionType.PROB multi_ex = [model.example("1 | a b c"), model.example("2 | a b c")] prediction = model.predict(multi_ex) assert isinstance(prediction, float) del model