histocartography.ml.models.cell_graph_model module¶
Summary¶
Classes:
Cell Graph Model. |
- class CellGraphModel(gnn_params: dict, classification_params: dict, node_dim: int, **kwargs)[source]¶
Bases:
histocartography.ml.models.base_model.BaseModel
Cell Graph Model. Apply a GNN at the cell graph level.
- __init__(gnn_params: dict, classification_params: dict, node_dim: int, **kwargs) → None[source]¶
CellGraphModel model constructor
- Parameters
gnn_params – (dict) GNN configuration parameters.
classification_params – (dict) classification configuration parameters.
node_dim (int) – Cell node feature dimension.
- forward(graph: Union[dgl.graph.DGLGraph, Tuple[None._VariableFunctions.tensor, None._VariableFunctions.tensor]]) → None._VariableFunctions.tensor[source]¶
Foward pass.
- Parameters
graph (Union[dgl.DGLGraph, Tuple[torch.tensor, torch.tensor]]) – Cell graph to process.
- Returns
Model output.
- Return type
torch.tensor
Reference¶
If you use histocartography in your projects, please cite the following:
@inproceedings{pati2021,
title = {Hierarchical Graph Representations for Digital Pathology},
author = {Pushpak Pati, Guillaume Jaume, Antonio Foncubierta, Florinda Feroce, Anna Maria Anniciello, Giosuè Scognamiglio, Nadia Brancati, Maryse Fiche, Estelle Dubruc, Daniel Riccio, Maurizio Di Bonito, Giuseppe De Pietro, Gerardo Botti, Jean-Philippe Thiran, Maria Frucci, Orcun Goksel, Maria Gabrani},
booktitle = {https://arxiv.org/pdf/2102.11057},
year = {2021}
}