cell_attn.model_definition_utils

Model definitions for Cell Movement Attention Networks

Classes

velocity_predictor	End to end function for predicting the next displacement of a focal cell.
pairwise_function_velocity	Pairwise interaction function definition.
weight_function	Weight function definition.

Module Contents

class cell_attn.model_definition_utils.velocity_predictor

Bases: torch.nn.Module

End to end function for predicting the next displacement of a focal cell.

interaction_network

weight_network

forward(x)

compute_weights(x)

class cell_attn.model_definition_utils.pairwise_function_velocity

Bases: torch.nn.Module

Pairwise interaction function definition.

Input: focal cell velocity (2d), nbh cell relative position (2d) and velocity (2d).

fc0

fc1

fc2

fc3

forward(x)

class cell_attn.model_definition_utils.weight_function

Bases: torch.nn.Module

Weight function definition.

Input: nbh cell relative position (2d).

fc0

fc1

fc2

fc3

forward(x)