Skip to content
Snippets Groups Projects
Select Git revision
1 result Searching

train.py

Blame
    • Code owners
    Assign users and groups as approvers for specific file changes. Learn more.
    quick_task.py 2.27 KiB
    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    14
    15
    16
    17
    18
    19
    20
    21
    22
    23
    24
    25
    26
    27
    28
    29
    30
    31
    32
    33
    34
    35
    36
    37
    38
    39
    40
    41
    42
    43
    44
    45
    46
    47
    48
    49
    50
    51
    52
    53
    54
    55
    56
    57
    58
    59
    60
    61
    62
    63 
    

    

    

    from tasks.task_hook import TaskHook


def quick_task(task_hook: TaskHook):
    def run_closeness(parameters, network, original_seeds=None):
        from .closeness_centrality import closeness_centrality

        def closeness_progress(progress, status):
            task_hook.set_progress(2 / 3 + 1 / 3 * progress, status)

        def closeness_set_result(result):
            result["network"]["edges"].extend(network["edges"])
            if original_seeds is not None:
                result['node_attributes']['is_seed'] = original_seeds
            task_hook.set_results(result)

        # Prepare intermediate hook
        closeness_task_hook = TaskHook(parameters,
                                       task_hook.data_directory,
                                       closeness_progress,
                                       closeness_set_result)

        # Run closeness centrality
        closeness_centrality(closeness_task_hook)

    def run_multi_steiner(parameters):
        from .multi_steiner import multi_steiner

        def ms_progress(progress, status):
            task_hook.set_progress(0 + 2 / 3 * progress, status)

        def ms_set_result(result):
            node_attributes = result.get("node_attributes", {})
            node_types = node_attributes.get("node_types", {})
            seeds = [seed for seed in result["network"]["nodes"] if node_types.get(seed) == 'protein']

            if len(seeds) == 0:
                task_hook.set_results({"network": {"nodes": [], "edges": []}})
                return
            parameters.update({
                "seeds": seeds,
                "result_size": 50,
                "hub_penalty": 0,
                "target": "drug",
                "include_non_approved_drugs": False
            })
            run_closeness(parameters, result["network"], result['node_attributes']['is_seed'])

        parameters["num_trees"] = 5
        parameters["tolerance"] = 5
        parameters["hub_penalty"] = 0.5

        # Prepare intermediate hook
        ms_task_hook = TaskHook(parameters,
                                task_hook.data_directory,
                                ms_progress,
                                ms_set_result)

        # Run multi_steiner
        multi_steiner(ms_task_hook)

    run_multi_steiner(task_hook.parameters)