Source code for tdc.benchmark_group.drugcombo_group
# -*- coding: utf-8 -*-
# Author: TDC Team
# License: MIT
from .base_group import BenchmarkGroup
[docs]class drugcombo_group(BenchmarkGroup):
"""create a drug combination benchmark group
Args:
path (str, optional): path to save/load benchmarks
"""
def __init__(self, path="./data"):
"""create a drug combination benchmark group"""
super().__init__(name="DrugCombo_Group", path=path, file_format="pkl")
[docs] def get_cell_line_meta_data(self):
import os
from ..utils.load import download_wrapper
from ..utils import load_dict
name = download_wrapper('drug_comb_meta_data', self.path, ['drug_comb_meta_data'])
return load_dict(os.path.join(self.path, name + '.pkl'))