graphorge.gnn_base_model.train.cross_validation

Cross-validation of Graph Neural Network model.

Functions

kfold_cross_validation

k-fold cross validation of Graph Neural Network model.

write_cross_validation_summary_file

Write summary data file for model cross-validation process.

Functions

kfold_cross_validation(cross_validation_dir, ...)

k-fold cross validation of Graph Neural Network model.

make_directory(directory[, is_overwrite])

Create a directory.

predict(dataset, model_directory[, model, ...])

Make predictions with Graph Neural Network model for given dataset.

train_model(n_max_epochs, dataset, ...[, ...])

Training of Graph Neural Network model.

write_cross_validation_summary_file(...)

Write summary data file for model cross-validation process.

write_summary_file(summary_directory[, ...])

Write summary data file with provided keyword-based parameters.