Python RasaModelData._split_by_label_ids Examples

Example #1

File: test_model_data.py Project: attgua/Geco

def test_split_data_by_label(model_data: RasaModelData):
    split_model_data = model_data._split_by_label_ids(
        model_data.data,
        model_data.get("intent", "ids")[0], np.array([0, 1]))

    assert len(split_model_data) == 2
    for s in split_model_data:
        assert len(set(s.get("intent", "ids")[0])) == 1

Example #2

def test_split_data_by_label(model_data: RasaModelData):
    split_model_data = model_data._split_by_label_ids(
        model_data.data, model_data.get("label", "ids")[0], np.array([0, 1])
    )

    assert len(split_model_data) == 2
    for s in split_model_data:
        assert len(set(s.get("label", "ids")[0])) == 1

    for key, attribute_data in split_model_data[0].items():
        for sub_key, features in attribute_data.items():
            assert len(features) == len(model_data.data[key][sub_key])
            assert len(features[0]) == 2