61ded5a18d28f8053d7547172fb2851b645c3dc4,alphapy/model.py,,save_predictions,#Any#Any#Any#,1184

Before Change

    // Save ranked predictions

    logger.info("Saving Ranked Predictions")
    pf["prediction"] = pd.Series(preds, index=pf.index)
    if model_type == ModelType.classification:
        pf["probability"] = pd.Series(probas, index=pf.index)
        pf.sort_values("probability", ascending=False, inplace=True)
    else:

After Change

        if found_pdate:
            probas = np.take(probas, pd_indices)
        prob_series = pd.Series(probas, index=pd_indices)
        df_prob = pd.DataFrame(prob_series, columns=["probability"])
        write_frame(df_prob, output_dir, output_file, extension, separator)

    // Save ranked predictions

Instances