model.py 文件源码

def x_label(feature_path, pred=False):
    X_list = []
    for each in feature_path:
        X = pd.read_csv(feature_paths.format(str(each)))
        X_list.append(X)
    X = pd.DataFrame(pd.concat(X_list, axis=0)).reset_index().drop('index', axis=1)
    if not pred:
        y = X[power_consumption].tolist()
        X = X.drop([record_date, user_id, power_consumption], axis=1)
        columns = X.columns
        X = X.values
        return X, y, columns
    else:
        X = X.drop([record_date, user_id], axis=1)
        columns = X.columns
        X = X.values
        return X, columns