modelData.py 文件源码

def test(feats_test, estimator=None, model='model.pkl'):
    """
    Evaluate the generated machine learning model on test data, and print a mean absolute error.
    @param estimator: The trained ML model/estimator
    @param feats_test: test features (obtained from data)
    """
    if estimator is None:
        estimator = joblib.load(cfg.PATH_RESOURCES+model)

    return estimator.predict(feats_test)