Skip to content

Tensors are just multi-dimensional arrays which generalizes scalars, vectors and matrices to higher dimensions. They are essential data structures in most machine learning frameworks and even more so in Deep Learning.

Notifications You must be signed in to change notification settings

42Kesara/Tensors-and-Graph-Neural-Networks-GNN-Implementation

Folders and files

NameName
Last commit message
Last commit date

Latest commit

Β 

History

5 Commits
Β 
Β 
Β 
Β 
Β 
Β 

Repository files navigation

πŸ“Š Tensors and Graph Neural Networks (GNNs) - Deep Dive into Graph-based Learning Welcome to our project on Tensors and Graph Neural Networks (GNNs)! This repository contains our group assignment for the Tensors and Graphs course, where we explore the theory, implementation, and applications of tensors and GNNs in machine learning.

πŸš€ Project Overview Tensors serve as the fundamental data structures for representing high-dimensional data in machine learning, while Graph Neural Networks (GNNs) allow us to learn from graph-structured data effectively. We apply GNNs to analyze relationships in the Cora dataset of scientific publications.

πŸ” Key Features Tensor Operations: Essential operations with tensors, using PyTorch and TensorFlow. Graph Neural Network Implementation: Implementation of key GNN architectures such as Graph Convolutional Networks (GCNs) and Graph Attention Networks (GATs). Cora Dataset Analysis: Node classification and link prediction tasks on the Cora dataset, utilizing GNNs for insightful analysis.

πŸ› οΈ Technologies Libraries: PyTorch, TensorFlow Data: Cora citation dataset with node-level and edge-level features Core Models: GCN, GAT, GraphSAGE, GIN

πŸ“‚ Repository Structure plaintext Copy code

πŸ“¦ GNN-Project β”œβ”€β”€ πŸ“ data/ # Cora dataset and preprocessing scripts β”œβ”€β”€ πŸ“ models/ # GNN model architectures β”œβ”€β”€ πŸ“ results/ # EDA, results, and model outputs β”œβ”€β”€ πŸ“„ README.md # Project documentation └── πŸ“„ Report.pdf # Detailed project report

πŸ“ˆ Methodology Data Processing: Loading and exploring the Cora dataset. Model Development: Implementing GNN architectures for node classification and link prediction. Training & Evaluation: Training GNNs with tensor operations and analyzing their performance.

πŸ“Š Results Our experiments show that: GCNs perform well on node classification tasks. GATs excel in adaptive weighting, improving performance on heterogeneous graphs. For details, check out our full Report.

πŸ“Š Results Screenshot 1 2 3 5 6

πŸ“… Future Work In future iterations, we plan to: Implement dynamic graphs for real-time data updates. Explore GNN scalability on larger datasets. Apply GNNs to other domains such as molecular chemistry and social network analysis.

πŸ‘₯ Team Kesara Lakpriya G.M. Chathura Samarajeewa Y.M. Weerathunga

πŸ“œ References Deep Learning with PyTorch, by Stevens et al. Graph Neural Networks: Foundations, Frontiers, and Applications, by Wu et al. 🌟 Contributing Want to contribute? Fork the repository, submit issues, or open a pull request. We'd love to hear your thoughts and improvements!

About

Tensors are just multi-dimensional arrays which generalizes scalars, vectors and matrices to higher dimensions. They are essential data structures in most machine learning frameworks and even more so in Deep Learning.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors