Python _check_confidence_setting Examples

Example #1

File: test_train_utils.py Project: praneethgb/rasa

def test_confidence_loss_settings(component_config: Dict[Text, Any],
                                  raises_exception: bool):
    component_config[SIMILARITY_TYPE] = INNER
    if raises_exception:
        with pytest.raises(InvalidConfigException):
            train_utils._check_confidence_setting(component_config)
    else:
        train_utils._check_confidence_setting(component_config)

Example #2

File: test_train_utils.py Project: praneethgb/rasa

def test_confidence_similarity_settings(component_config: Dict[Text, Any],
                                        raises_exception: bool):
    component_config[LOSS_TYPE] = SOFTMAX
    if raises_exception:
        with pytest.raises(InvalidConfigException):
            train_utils._check_confidence_setting(component_config)
    else:
        train_utils._check_confidence_setting(component_config)

Example #3

File: test_train_utils.py Project: zoovu/rasa

def test_confidence_renormalization_settings(component_config: Dict[Text, Any],
                                             raises_exception: bool):
    component_config[LOSS_TYPE] = CROSS_ENTROPY
    if raises_exception:
        with pytest.raises(InvalidConfigException):
            train_utils._check_confidence_setting(component_config)
    else:
        train_utils._check_confidence_setting(component_config)

Example #4

File: test_train_utils.py Project: praneethgb/rasa

def test_warn_deprecated_model_confidences():

    component_config = {MODEL_CONFIDENCE: LINEAR_NORM}

    with pytest.warns(FutureWarning):
        train_utils._check_confidence_setting(component_config)