Datawhale AI夏令营-跑通baseline的第一天

  1. 导入库:首先,代码导入了需要用到的库,包括 pandas(用于数据处理和分析),scikit-learn(机器学习库),rdkit(化学信息工具)。

  2. 读取数据:代码通过使用 pd.read_csv 函数从文件中读取训练集和测试集数据。

  3. 使用Morgan分子指纹建模SMILES

- 这个过程需要调用rdkit的相关模块。然后将Reactant1,Reactant2,Product,Additive,Solvent字段的向量拼接到一起,组成一个更长的向量。

使用随机森林预测结果

- 这里直接调用sklearnRandomForestRegressor模块实例化一个随机森林模型,并对n_estimators等重要参数进行指定。最后使用model.fit(x, y)训练模型。模型保存在本地'./random_forest_model.pkl'

加载模型进行预测,并将保存结果文件到本地:

- pkl文件直接使用pickle.load()加载,然后使用model.predict(x)进行预测。预测的结果保存为比赛官方指定的文件格式。

感想:

本以为会很复杂,不过跟着教程一步一步来,很顺利的就跑通了,加油!

  • 3
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值