药物相互作用预测结束后怎么把测试集输出并保存
时间: 2024-03-20 20:42:21 浏览: 52
在药物相互作用预测结束后,你可以使用Python的pandas库来将测试集输出并保存为CSV文件。以下是一些代码示例,可以根据你的具体需求进行修改:
``` python
import pandas as pd
# 假设你的测试集数据存储在test_data变量中
# 将预测结果存储在predicted_labels变量中
# 创建一个DataFrame对象,将测试集和预测结果合并
test_results = pd.DataFrame({
'compound_1': test_data['compound_1'],
'compound_2': test_data['compound_2'],
'predicted_label': predicted_labels
})
# 将DataFrame对象保存为CSV文件
test_results.to_csv('test_results.csv', index=False)
```
这段代码中,我们使用了pandas的DataFrame对象来将测试集和预测结果合并,并创建了一个新的DataFrame对象`test_results`。然后,我们使用to_csv()方法将`test_results`保存为CSV文件,其中`index=False`参数表示不保存DataFrame的行索引。你可以将文件名`test_results.csv`替换为你想要的文件名和路径。
阅读全文