586b49b07574f7812101248c94d73ef1fb2a5c12,mltsp/tests/test_predict.py,,test_add_to_predict_results_dict,#,37

Before Change

def test_add_to_predict_results_dict():
    Test add data to predict results dict
    for suffix in ["targets.npy", "features.csv"]:
        shutil.copy(
            pjoin(DATA_PATH, "test_%s" % suffix),
            pjoin(cfg.FEATURES_FOLDER, "TEST001_%s" % suffix))
    results_dict = {}
    pred.add_to_predict_results_dict_classification_proba(
        results_dict, [[0.2, 0.5, 0.3]], "TT.dat",
        [1, 2, 3], {"f1": 2}, "TEST001", 5)

After Change

    Test add data to predict results dict
    results_dict = {}
    preds = pd.Series({"class1": 0.2, "class2": 0.5, "class3": 0.3}, name=0)
    ts_data = np.array([[1., 2., 3.], [4., 5., 6.]])
    predict.add_to_predict_results_dict_classification_proba(
        results_dict, preds, "TT.dat",
        ts_data, {"f1": 2}, "TEST001", 5)

Instances