predict.py 文件源码

def predict(all_features, categorical_features, connection_string, filename):
    engine = create_engine(connection_string)
    model = joblib.load(filename)
    print model

    query = """SELECT * FROM connection_features"""
    df = pd.read_sql_query(query, con=engine, index_col=['departurestop', 'departuredate', 'route'])
    df.index.levels[0].name = 'stationfrom'
    df.index.levels[1].name = 'date'
    df.index.levels[2].name = 'vehicle'
    df = df.reset_index()

    predicted = model.predict(df[all_features])
    print predicted