medmodels.treatment_effect.temporal_analysis#

This module contains functions for temporal analysis of treatment effects.

Functions

find_reference_edge(medrecord, node_index, ...)

Determines the reference edge containing the first/last last exposure.
find_reference_edge(medrecord, node_index, connected_group, reference, time_attribute='time')[source]#

Determines the reference edge containing the first/last last exposure.

This function returns the edge containing the first/last exposure (depending on the reference argument) of a node to any other node in the connected_group. This method is crucial for analyzing the temporal sequence of treatments and outcomes.

Parameters:

  • medrecord (MedRecord) – An instance of the MedRecord class containing medical data.

  • node_index (NodeIndex) – The node for which to determine the reference exposure edge.

  • connected_group (Group) – The group of nodes that are connected to the node.

  • reference (Literal["first", "last"]) – The reference point for the exposure time. Options include “first” and “last”. If “first”, the function returns the earliest exposure edge. If “last”, the function returns the latest exposure edge.

  • time_attribute (MedRecordAttribute, optional) – The attribute in the edge that contains the time information. Defaults to “time”.

Return type:

int

Returns:

EdgeIndex – The edge index of the reference exposure.

Raises:

ValueError – If no edge with that time attribute or datetime datatype found for the node in this MedRecord.

Example

This function returns the edge containing the timestamp of the last exposure to any medication in the group “medications” for subject “P1”.

find_reference_edge(
    medrecord,
    node_index="P1",
    connected_group="medications",
    reference="last",
)
On this page