ncem.estimators.EstimatorEDncem.predict_embedding_any

EstimatorEDncem.predict_embedding_any(img_keys, node_idx, batch_size=1)

Predict embedding on any given data set.

Parameters

• img_keys – Image keys.

• node_idx – Nodes indices.

• batch_size (int) – Number of samples. If unspecified, it will default to 1.

Return type

eval_dict