用户注册



邮箱:

密码:

用户登录


邮箱:

密码:
记住登录一个月忘记密码?

发表随想


还能输入:200字
云代码 - python代码库

circus

2019-12-12 作者: 半陶举报

[python]代码库

import pandas as pd

def run(train_df, trust_df, userinfo_df, test_df):
    """
    参赛者代码。
    必须重写这个run()方法作为入口函数。
    输入参数必须为给出的3个训练数据集df以及测试数据集df(如果有部分训练数据并没有用到,也务必传入),并按train_df, trust_df, userinfo_df, test_df顺序传入。
    方法里需要训练模型(模型不需要保存),然后直接执行预测,得到测试集的预测结果,并输出预测结果的df
    输出的结果中必须包含Item、User、Prediction这3列,且Prediction的值仅可为0/1。
    """
    print("log here ....")
    pred = test_df.apply(lambda row: 1, axis=1)
    out_test_df = test_df.copy(deep=True)
    out_test_df["Prediction"] = pred
    return out_test_df

if __name__ == '__main__':
    """
    以下代码可作为参赛者本地调参使用,与线上比赛系统无关。
    线上比赛系统直接调用run()方法作为入口函数。
    """

    # 数据从文件读入参考该部分代码。
    train_file = 'train.csv'
    trust_file = 'trust.csv'
    userinfo_file = 'user_info.csv'
    test_file = 'test.csv'
    predict_file = 'results_tmp.csv'

    train_df = pd.read_csv(train_file, engine='python', encoding='utf-8')
    trust_df = pd.read_csv(trust_file, engine='python', encoding='utf-8')
    userinfo_df = pd.read_csv(userinfo_file, engine='python', encoding='utf-8')
    test_df = pd.read_csv(test_file, engine='python', encoding='utf-8')

    # 方法调用
    result_df = run(train_df, trust_df, userinfo_df, test_df)
    result_df.to_csv(predict_file, index=False)


网友评论    (发表评论)


发表评论:

评论须知:

  • 1、评论每次加2分,每天上限为30;
  • 2、请文明用语,共同创建干净的技术交流环境;
  • 3、若被发现提交非法信息,评论将会被删除,并且给予扣分处理,严重者给予封号处理;
  • 4、请勿发布广告信息或其他无关评论,否则将会删除评论并扣分,严重者给予封号处理。


扫码下载

加载中,请稍后...

输入口令后可复制整站源码

加载中,请稍后...