ncem.estimators.EstimatorNoGraph.predict

EstimatorNoGraph.predict(batch_size: int = 1)Tuple[Dict[str, numpy.ndarray], Dict[str, numpy.ndarray]]

Return observed labels and full predictions (including scale model) grouped exactly as in nodes_idx_test.

Parameters

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

Returns

Return type

predict