# 🎨 Mixture-Density-Nets
A small PyTorch library for Mixture Density Networks.
# Install
simply run
``pip install mixture-density-nets``
# Example
```py
from mixture_density_nets import MDN, MDDistribution
# ....
mdn = MDN(in_dim, out_dim, n_components)
# ....
mu, sigma, lambda_ = mdn(net(input_data))
dist = MDDistribution(mu, sigma, lambda_)
loss = dist.nll(targets).mean()
# ...
samples, clusters = dist.sample(n=20) # draw 20 samples
```
For a more thorough example see [example.ipynb](example.ipynb).
Raw data
{
"_id": null,
"home_page": "https://github.com/romue404/mixture-density-nets",
"name": "Mixture-Density-Nets",
"maintainer": "",
"docs_url": null,
"requires_python": "",
"maintainer_email": "",
"keywords": "artificial intelligence,pytorch,mixture density network",
"author": "Robert M\u00fcller",
"author_email": "robert.mueller1990@googlemail.com",
"download_url": "https://files.pythonhosted.org/packages/a0/6e/673a54c7766af0d19f21bd42cbc8af98d0118789400f192dfed652b27acb/Mixture-Density-Nets-0.1.1.tar.gz",
"platform": null,
"description": "# \ud83c\udfa8 Mixture-Density-Nets\nA small PyTorch library for Mixture Density Networks.\n\n# Install\nsimply run \n``pip install mixture-density-nets``\n\n# Example\n```py\nfrom mixture_density_nets import MDN, MDDistribution\n# ....\nmdn = MDN(in_dim, out_dim, n_components)\n# ....\nmu, sigma, lambda_ = mdn(net(input_data))\ndist = MDDistribution(mu, sigma, lambda_)\nloss = dist.nll(targets).mean()\n\n# ...\nsamples, clusters = dist.sample(n=20) # draw 20 samples\n```\nFor a more thorough example see [example.ipynb](example.ipynb).\n",
"bugtrack_url": null,
"license": "MIT",
"summary": "A small PyTorch library for Mixture Density Networks.",
"version": "0.1.1",
"project_urls": {
"Homepage": "https://github.com/romue404/mixture-density-nets"
},
"split_keywords": [
"artificial intelligence",
"pytorch",
"mixture density network"
],
"urls": [
{
"comment_text": "",
"digests": {
"blake2b_256": "3f701abcae4f795b718dff08d80aa99cbeca0d989b7985ccdcfef06c84f110cc",
"md5": "0622c87b4cc1b4c554199f5782df7ee1",
"sha256": "dac5a95f589be5d88318a744e0765fbbd0af19dd82de0a93b7cd1165ba7a261c"
},
"downloads": -1,
"filename": "Mixture_Density_Nets-0.1.1-py3-none-any.whl",
"has_sig": false,
"md5_digest": "0622c87b4cc1b4c554199f5782df7ee1",
"packagetype": "bdist_wheel",
"python_version": "py3",
"requires_python": null,
"size": 3590,
"upload_time": "2023-09-10T11:10:11",
"upload_time_iso_8601": "2023-09-10T11:10:11.085609Z",
"url": "https://files.pythonhosted.org/packages/3f/70/1abcae4f795b718dff08d80aa99cbeca0d989b7985ccdcfef06c84f110cc/Mixture_Density_Nets-0.1.1-py3-none-any.whl",
"yanked": false,
"yanked_reason": null
},
{
"comment_text": "",
"digests": {
"blake2b_256": "a06e673a54c7766af0d19f21bd42cbc8af98d0118789400f192dfed652b27acb",
"md5": "740ab55914e76ec70cf96eca342e35d5",
"sha256": "a14ef9b5cbaeb46f579334cd3b15417d5ebb3482e4142d9ce70a135fb9d593b9"
},
"downloads": -1,
"filename": "Mixture-Density-Nets-0.1.1.tar.gz",
"has_sig": false,
"md5_digest": "740ab55914e76ec70cf96eca342e35d5",
"packagetype": "sdist",
"python_version": "source",
"requires_python": null,
"size": 3096,
"upload_time": "2023-09-10T11:10:12",
"upload_time_iso_8601": "2023-09-10T11:10:12.759685Z",
"url": "https://files.pythonhosted.org/packages/a0/6e/673a54c7766af0d19f21bd42cbc8af98d0118789400f192dfed652b27acb/Mixture-Density-Nets-0.1.1.tar.gz",
"yanked": false,
"yanked_reason": null
}
],
"upload_time": "2023-09-10 11:10:12",
"github": true,
"gitlab": false,
"bitbucket": false,
"codeberg": false,
"github_user": "romue404",
"github_project": "mixture-density-nets",
"travis_ci": false,
"coveralls": false,
"github_actions": true,
"lcname": "mixture-density-nets"
}
JSON
