34b74a059dc70f49c6033d9593ca6f6b1979c058,alphapy/model.py,,predict_best,#Any#,750

Before Change

    // Record support vector for any recursive feature elimination

    if rfe and model.support[best_tag]:
        pass

    // Return the model with best estimator and predictions.

    end_time = datetime.now()
    time_taken = end_time - start_time
    logger.info("Best Model Selection Complete: %s", time_taken)

After Change

    // Record support vector for any recursive feature elimination

    if rfe:
        try:
            if model.support[best_algo]:
                logger.info("Saving RFE Support")
                full_path = SSEP.join([directory, "model", "features_support_rfe.pkl"])
                joblib.dump(model.support[best_algo], full_path)
        except:
            // no RFE support for best algorithm
            pass

    // Return the model with best estimator and predictions.

    end_time = datetime.now()
    time_taken = end_time - start_time
    logger.info("Best Model Selection Complete: %s", time_taken)

Instances