search

Recode-Hive / machine-learning-repos

A curated list of awesome machine learning frameworks, libraries and software (by language). I
MIT License
45 stars 125 forks source link

Word Embedding System Using Pytorch #495

Closed ubaid2751 closed 4 days ago

ubaid2751 commented 4 days ago

Summary of Changes

This pull request introduces a simple neural network model to create word embeddings using a linear transformation. The model is implemented using PyTorch and trained on a small corpus of text. Key changes include:

->Tokenization and one-hot encoding of the text corpus. ->Definition of the WordEmbeddingWithLinear model. ->Training procedure for the model. ->Visualization of word embeddings before and after training.

Motivation and Context

The motivation behind this change is to provide a straightforward example of generating word embeddings using a basic neural network model. This can be a foundational step for more complex natural language processing tasks. The context of this change is within the scope of demonstrating basic machine learning concepts and techniques using PyTorch.

github-actions[bot] commented 4 days ago

Thank you for submitting your pull request! 🙌 We'll review it as soon as possible. In the meantime, please ensure that your changes align with our CONTRIBUTING.md. If there are any specific instructions or feedback regarding your PR, we'll provide them here. Thanks again for your contribution! 😊