autofragment.algorithms.ml_interface

Machine learning interface for fragmentation (Future).

Functions

extract_node_features(graph)	Extract per-node features for ML.
generate_training_data(systems, partitions)	Generate training data from expert fragmentations.

Classes

GNNFragmentationModel([hidden_dim, n_layers])	Graph Neural Network for fragmentation prediction.
MLFragmentationModel()	Abstract interface for ML-based fragmentation.