ncem.estimators.EstimatorGraph.split_data_given

EstimatorGraph.split_data_given(img_keys_test, img_keys_train, img_keys_eval, nodes_idx_test, nodes_idx_train, nodes_idx_eval)

Split data by given partition.

Parameters

• img_keys_test – Test image keys.

• img_keys_train – Train image keys.

• img_keys_eval – Evaluation image keys.

• nodes_idx_test – Test node indices.

• nodes_idx_train – Train node indices.

• nodes_idx_eval – Evaluation node indices.