AI4S Cup学习赛-中枢神经系统药物研发:药物筛选与优化

本文介绍: Bo hr ium 案例广场 (dp.t e ch)中枢神经系统类疾病长期以来存在着重要的临床未满足需求。据统计，在当前人口老龄化趋势下，阿兹海默（AD）、帕金森病（PD）等神经退行性疾病和脑癌、中风等疾病的治疗费用将达到数万亿美元，由此产生巨额的医疗保健支出。因此，获得高质量、有效的中枢神经系统药物在当前被普遍视为具有重要的科学和商业价值。随着AI fo r sci ence的兴起，AI可以在大规模化合物库中筛选具有潜在治疗作用的药物，基于计算机模拟的方法预测药物的药理活性、毒性和药代动力学特性。

链接：Bohrium 案例广场 (dp.tech)

中枢神经系统类疾病长期以来存在着重要的临床未满足需求。据统计，在当前人口老龄化趋势下，阿兹海默（AD）、帕金森病（PD）等神经退行性疾病和脑癌、中风等疾病的治疗费用将达到数万亿美元，由此产生巨额的医疗保健支出。因此，获得高质量、有效的中枢神经系统药物在当前被普遍视为具有重要的科学和商业价值。

随着AI fo r sci e nc e的兴起，AI可以在大规模化合物库中筛选具有潜在治疗作用的药物，基于计算机模拟的方法预测药物的药理活性、毒性和药代动力学特性。通过这种方式，AI可以筛选出最有前景的药物候选者，并优化其化学结构，以提高安全性和有效性。

本次学习赛将提供一批分子信息数据，选手需要以AI方法对分子是否可作为CNS（Centr al Ne r vous Sy ste m）药物进行检测，实现AI助力下的CNS药物筛选和研发工作。

中枢神经系统(CNS)包括大脑和脊髓，负责处理和调控身体的各种功能。非中枢神经系统（non-CNS）则包括神经元以外的组织，如内分泌系统和免疫系统等。

由于血脑屏障及其上各类转运体的存在，中枢神经系统用药通常需要满足一些特定的特征。药物化学家会根据特定的指标来判断一个药物是否具有成为中枢神经系统药物的潜力，如分子量和拓扑极性表面积（TPSA）不能过大，N原子和极性H原子数量不能过多，溶剂可及表面需在特定范围内等。这些特征为药物发现和优化阶段中判断分子作为CNS药物的潜力提供了启示。

# 需要挂载 bohr CNS 数据集
DIR_PATH = '/bohr/ai4scup-cns-5zkz/v3/'
!pip install lightgbm numpy pandas rdkit scikit-learn

import lightgbm as lgb
import numpy as np
import pandas as pd

from rdkit import Chem  # 导入RDKit中的Chem模块，用于分子对象转换
# 从RDKit中导入GetMorganFingerprintAsBitVect函数，用于生成分子指纹（位向量转换）
from rdkit.Chem.rdMolDescriptors import GetMorganFingerprintAsBitVect  

from sklearn.model_selection import train_test_split  # 从scikit-learn中导入train_test_split函数，用于拆分数据集
from sklearn.metrics import fbeta_score  # 从scikit-learn中导入fbeta_score函数，用于 F2 Score 计算

raw_data = pd.read_csv(f"{DIR_PATH}/mol_train.csv")
test_data = pd.read_csv(f"{DIR_PATH}/mol_test.csv")

# 拆分训练数据为训练集与验证集，验证集占比 20%，设定固定随机种子
train_data, valid_data = train_test_split(
    raw_data, test_size=0.2, random_state=hash("Datawhale") % 2023
)

def smile2fingerprint(smile: str):
    """将 SMILE 分子式表示为指纹数据

    参数:
        smile (string): SMILE 分子式

    返回:
        fp (Explict BitVect): 分子式的 Morgan 指纹位向量
    """
    molecular = Chem.MolFromSmiles(smile)  # 将字符串转换为分子式对象
    finger_print = GetMorganFingerprintAsBitVect(molecular, 5, nBits=1024)  # 获得分子式的 Morgan 指纹位向量
    return finger_print

# 批量将位向量转换为特征矩阵，形状为 (n, 1024)，n 代表数据个数，1024 在位向量转换时指定
train_X = np.array([smile2fingerprint(smile) for smile in train_data["SMILES"]])
valid_X = np.array([smile2fingerprint(smile) for smile in valid_data["SMILES"]])
test_X = np.array([smile2fingerprint(smile) for smile in test_data["SMILES"]])

# 将数据特征矩阵转换为 LightGBM 指定格式，(特征向量，对应标签)
lgb_train = lgb.Dataset(train_X, label=train_data["TARGET"])
lgb_valid = lgb.Dataset(valid_X, label=valid_data["TARGET"])

# 设定 LightGBM 训练参，查阅参数意义：https://lightgbm.readthedocs.io/en/latest/Parameters.html
lgb_params = {
    "objective": "binary",  # 指定任务类别为二分类
    "seed": hash("Datawhale") % 2023,  # 设定随机种子
    "verbose": -1,  # 禁用输出（可选）
}

# 训练模型，参数依次为：导入模型设定参数、导入训练集、设定模型迭代次数（100）、导入验证集
model = lgb.train(lgb_params, lgb_train, num_boost_round=100, valid_sets=lgb_valid)

threshold = 0.5 # 模型输出的是类别概率，设定概率的判断阙值

# 用验证集进行模型预测（选择训练中最好的一次）
valid_pred = model.predict(valid_X, num_iteration=model.best_iteration)
# 生成预测标签结果，如果概率大于阈值则为 1，否则为 0
valid_result = [1 if x > threshold else 0 for x in valid_pred]
# 计算验证集 F2 Score 分数
valid_score = fbeta_score(valid_data["TARGET"], valid_result, beta=2)
print(f"Valid Score: {valid_score}")

# 预测测试集数据并获得预测结果
pred = model.predict(test_X, num_iteration=model.best_iteration)
result = [1 if x > threshold else 0 for x in pred]

submission = pd.DataFrame()
submission["SMILES"] = test_data["SMILES"]
submission["TARGET"] = result
submission.to_csv("./submission.csv", index=False)