v0.6.0
What's Changed
- Typo fix is_writen -> is_written by @copybara-service in #304
- For examples/notebooks, install the latest release (not release candidate). by @copybara-service in #306
- Add OGBN-MAG colab links by @copybara-service in #305
- Partially re-enable some tests for TF2.9. by @copybara-service in #316
- Expose integrated gradient function to enable building custom exporter with it. by @copybara-service in #319
- Add causal masking when training with labels. by @copybara-service in #321
- Fixes double-install of tf-nightly and tensorflow in nightly test script. by @copybara-service in #323
- GCN model: Different normalizations of the adjacency matrix is added. The value of 'degree_normalization' controls how the edges are normalized. The default value of 'degree_normalization' is set to 'in_out'. Note that the default value gives a different output for asymmetric graphs compared to the previous default value. To reinstate the old behaviour of the code for asymmetric graphs, set 'degree_normalization' to 'in_in'. The 'normalize' flag is removed. To make sure the edges are not normalized, set the 'degree_normalization' to 'none'. by @copybara-service in #326
- For eventual multi-task support: by @copybara-service in #335
- This change is a no-op by @copybara-service in #262
- Raise exception if a key has been provided, but no dictionary is available for
ResidualNextState
. by @copybara-service in #353 - Hooks for HGT hyperparameter tuning by @copybara-service in #356
- Update TF-GNN models to best practices: by @copybara-service in #358
- Remove the unnecessary epsilon parameter. by @copybara-service in #363
- Call validate_graph_tensor_spec_for_readout() from validate_schema(). by @copybara-service in #364
- Remove the function random_graph_tensor_from_schema(). by @copybara-service in #367
- Fix typo: s/contined/continued/ by @copybara-service in #379
- Merge DGI layers with others. by @copybara-service in #385
- Add code for upcoming generalized pool() and broadcast() operations. by @copybara-service in #384
- Add more granular shuffling. by @copybara-service in #387
- Unit-test broadcast_node_to_edges() for Adjacency and HyperAdjacency. by @copybara-service in #394
- Expose
RunResult
. by @copybara-service in #395 - Merge DGI layers with others. by @copybara-service in #402
- Updates TF version for newest stable tests to 2.12 by @copybara-service in #410
- Allow passing collection of receiver tags to ConvGNNBuilder by @copybara-service in #412
- Fix the behavior of shuffle_tensor for 0 shapes. This happens with make_empty_feature and shuffling rate not in {0, 1}. by @copybara-service in #416
- Correctly handle auxiliary graph pieces in convert_to_line_graph by @copybara-service in #422
- TFLite tests for vanilla_mpnn and mt_albis by @copybara-service in #425
- Allows specialization how composite tensors are converted for the bulk executor. by @copybara-service in #432
- Silence some pytype errors. by @copybara-service in #439
- Update
attribution.py
testing: the preprocessing model created by orchestration returns aSequence[GraphTensor]
. by @copybara-service in #442 - Add Keras wrappers
ReadoutNamedIntoFeature
andAddReadoutFromFirstNode
by @copybara-service in #452 - Adjust
maxval += 1
in the case of an integer random counterfactual. by @copybara-service in #457 - Switch tfgnn.pool() to implementation v2 and remove v1. by @copybara-service in #464
- Enable test case: subgraph sampling model, saving and loading, as tf.keras.Model. by @copybara-service in #468
- Utils to read unigraph-formatted data into sampler v2 input format and a driver program used like
graph_sampler.py
by @copybara-service in #469 - Align context indices dtype resolution between GraphTensor and GraphTensorSpec by @copybara-service in #477
- Redefines
indices_dtype
as two concepts: by @copybara-service in #486 - Add convenience function. by @copybara-service in #495
- Fix type annotation:
SimpleConv(..., receiver_feature=None)
is supported by @copybara-service in #506 - Fixes type annotation syntax in runner link prediction by @copybara-service in #511
- Disable test that fails for unrelated reasons. by @copybara-service in #518
- Sampler pieces are removed from
tfgnn
core package. by @copybara-service in #519 - Re-enables edge sampling tests. by @copybara-service in #529
- Implement the Coherence metric. by @copybara-service in #533
- Fix for inconsistent ragged row partitions of edges and indices to select unique edges. by @copybara-service in #535
Full Changelog: v0.5.1...v0.6.0