From d0d0d1e002749ad3c4c8fa88507ebcb401721aac Mon Sep 17 00:00:00 2001
From: Sree Vidya <sreevidyapatamsetti@gmail.com>
Date: Tue, 23 Jul 2024 17:56:30 +0530
Subject: [PATCH] Add files via upload

---
 AI_CHATBOT/README.md     |   4 ++
 AI_CHATBOT/chat.py       |  53 +++++++++++++++
 AI_CHATBOT/data.pth      | Bin 0 -> 5674 bytes
 AI_CHATBOT/intents.json  | 138 +++++++++++++++++++++++++++++++++++++++
 AI_CHATBOT/model.py      |  19 ++++++
 AI_CHATBOT/nitkutils.py  |   0
 AI_CHATBOT/nltk_utils.py |  23 +++++++
 AI_CHATBOT/train.py      | 108 ++++++++++++++++++++++++++++++
 8 files changed, 345 insertions(+)
 create mode 100644 AI_CHATBOT/README.md
 create mode 100644 AI_CHATBOT/chat.py
 create mode 100644 AI_CHATBOT/data.pth
 create mode 100644 AI_CHATBOT/intents.json
 create mode 100644 AI_CHATBOT/model.py
 create mode 100644 AI_CHATBOT/nitkutils.py
 create mode 100644 AI_CHATBOT/nltk_utils.py
 create mode 100644 AI_CHATBOT/train.py

diff --git a/AI_CHATBOT/README.md b/AI_CHATBOT/README.md
new file mode 100644
index 0000000..4880f4d
--- /dev/null
+++ b/AI_CHATBOT/README.md
@@ -0,0 +1,4 @@
+# AI_CHATBOT
+It is an Ai chatbot developed using natural language toolkit NLTK, pytorch
+I have developed this bot for a Petari website 
+Feel free to update the intents.json to make the domain  of the chatbot wider.
diff --git a/AI_CHATBOT/chat.py b/AI_CHATBOT/chat.py
new file mode 100644
index 0000000..333859d
--- /dev/null
+++ b/AI_CHATBOT/chat.py
@@ -0,0 +1,53 @@
+import random
+import json
+
+import torch
+
+from model import NeuralNet
+from nltk_utils import bag_of_words, tokenize
+
+device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+
+with open('intents.json', 'r') as json_data:
+    intents = json.load(json_data)
+
+FILE = "data.pth"
+data = torch.load(FILE)
+
+input_size = data["input_size"]
+hidden_size = data["hidden_size"]
+output_size = data["output_size"]
+all_words = data['all_words']
+tags = data['tags']
+model_state = data["model_state"]
+
+model = NeuralNet(input_size, hidden_size, output_size).to(device)
+model.load_state_dict(model_state)
+model.eval()
+
+bot_name = "Snapitizer"
+print("Let's chat! (type 'quit' to exit)")
+while True:
+    # sentence = "do you use credit cards?"
+    sentence = input("You: ")
+    if sentence == "quit":
+        break
+
+    sentence = tokenize(sentence)
+    X = bag_of_words(sentence, all_words)
+    X = X.reshape(1, X.shape[0])
+    X = torch.from_numpy(X).to(device)
+
+    output = model(X)
+    _, predicted = torch.max(output, dim=1)
+
+    tag = tags[predicted.item()]
+
+    probs = torch.softmax(output, dim=1)
+    prob = probs[0][predicted.item()]
+    if prob.item() > 0.75:
+        for intent in intents['intents']:
+            if tag == intent["tag"]:
+                print(f"{bot_name}: {random.choice(intent['responses'])}")
+    else:
+        print(f"{bot_name}: I do not understand...")
\ No newline at end of file
diff --git a/AI_CHATBOT/data.pth b/AI_CHATBOT/data.pth
new file mode 100644
index 0000000000000000000000000000000000000000..c3fea7a6a57575a2ad8924c1ee133eab7cbb79d7
GIT binary patch
literal 5674
zcmb7I30PCd77ic~Q9uM01QmCz$`()%?ioM?G$@FuxMCm|6AcNITtHM@0TmZqS(Vlu
z+^cox&c#+M?rYuk`4rr$*4ACDzI$)f0`Ik-?|$Eqb5G9ypMPdfW|)bVn>CV1q*BR$
zJ=T&I5~YGscnE*oQzmM{LhK|X8$L5kGqJ=it_crM)+uRCyn#_LG&V_0)M+$yBBR!6
z4OUTlC9S8Gq3T4&3Y&I|!Deo8jd_H|%RQY|k5@6cQ6i($C#u}zQyI0!U=^>Y6H?V0
zWjsS`4LW^%nzt1;cXNxg<Us~LQZb&!7Dk9wm`0~yh8Uez*fP$9kMYFPI5Yk;F(nl@
z9(cus!B&Ii61h|!kb|4Z1LP(-xamBVpv0QNHZj;%Wai~0s1*j>OvUF?S*VO$Uf536
zST1F7bCDt&+sjR|v4aubg263ga4TbPyg&EGu^xdpcH#+w7~I;ZL7Ib|c`3XG7b8g<
z2DgpD?Ti|{jT&6*QOIz6qlONjG;|a+bTYy_Gq_6(?rPK^`>esO9)S<;#uNB4_#1)1
zQqF7WZq(4jNYazR?lIWIs6l4b;8~Bt7klv(mJIe5DYCIl(BNZ)`!cv!4DOwpiTy+!
z@yRr!5DEwTi^JT<EsmcK2@c4_fnts{T5sTM15<JnL!2r9mBtH0CPu2i9Nsu669?CE
z$Z$v|?rU`LMJv(Mnp7N`iTjDJR6|0X74KDRQ&O3DgL*QJ!{h;CWvbLlC9O63!sSxY
zr%PqN#6`$0Mb!$8CO%!KR~m5tF?fKRsH>|1%f*^f;K(=&o?nreNT)D(V4S(|qDe?i
z#)FKuHiPd_lvsEQJ&mJ9D<Ol%gGFGXBAM1JaE!4s6<RzbPUuCVLa)TJpG<=a4;9H0
zRWwe;!^BvnA_ET>#VB<+&gjz7c!U_phpAKWNYPG8)oRsvlo&CdX7Ff{b-Yfe#A8GW
zf|-zk$BKTHB8|rJB2YzZG&(%a=%zI(SRrz#=nR}7CRV9&qS2*G$4YTL)R-2(QX6oR
z2%MmsNaOLMIZ>@uVwKU<@(8t<MZ>QQdOSgNYjoQ2c%n$D(J@#fQYF(k*=S^7t!U_y
zuuhEEX*Ft`BF3li!-2&j;;m#_%V51o&%ZIr*dTT^g<on?^_UUE^t3^zNo8=V2sh9)
zP7?tJm5xuHE&>cXzG78)l9-H9@IA>en*7M%$znL8QfMdQDMq(S!QiPPAH&a7=BEID
zmBZ7-fOH{X`X`gt;u&H@x=Nj>!ZVF-HP1ZjlgX!^EwW|kQgN2(<SjkUrbU}kj5pw%
zF?fz!-AbApCoJ3iqO0bAO&NHem_n$544f-=+&JAdo-YQD*V8nk=7)WO`1)z^40xf)
zDfZ5Q7tyKi(Q-?vo9(||UgRrWXwiSWN7zU_jeA5w21D~d99kb1(&N83iH4g-op+6z
zVKlUmF`K&UB>cXF_V*c^NP_>`l{`GdLa6^MU7gHE^IFY2cA$?HnZ5BjoV|Xm;)Obk
z#N_7z<GzThdH$HHUUihcSaB3i&Rz-opI3wGKx^XG-wLgY9S1Fb2!e0>hf!rgF|14O
zNw)vULdr$f7lxJhC4Q3^u$v$EB@<0UNgtcr<Z$60mf5_QYBc#HyK7_|xcr$*^mkeI
zO{Z6st#lUqdz(^LGUQGl_n;bJRxX4ajh<9Y9(D%~&nacIt>(h5TNmJM39z1<3aODr
z6RE?KP0?-VTz3A-pUI$$%h>n&y<oTNQee{nOC&A+J@9F(+ti0Gh3q1`66$REVkomq
zf~bcR$<5t0FmRNTs=nQk4Q(|V2AzBekBYO1uiOR|`q`u0*ZI)CUmm>BUZPes*$Jcf
zP-t|tloYOA4)noQU>bgjeYg55wWQZl;u2R2H{TV*GqXM<eH($C{;4oM;vCg;fH{<u
zo`=Dk9MHqJCdlsHd6@UgoH#qJB&!}hVVycVA-f(4Bz=bsY7u=ICXBK`Pi8s5wxS0x
zZ}(KlyO#$K&I~3;_wQjJkKw?d*-~+%dv|0K`zMJUZpH4B?gcKn7G}kmpk`hUY)NuU
z^fp*VTAEkE$KVhc>%9%Sgcp#iF5j~SDeffcSyAA+3umFS&lbpVv?NS$C3|3=loU*e
zB!eAlNpiR5RG4cGJnYMnjI!wm<^4K=iPVHVnN$`y_S{CW|APvAU1>^cW?CZHmPYo+
zzJgk)fx;L&WWS~}@%#QM1RR(}PK4j1jz>+SqM};?Nai0bvS>w0k5{v!hPQ^{3oBs8
zv4^BIx;1%Z-Gao7@guFG9HH!zE&TQfgX=p>SY|VZWCm_ztzR!91;IVJ@M?<m`gcCq
zRm_0LHbszIBZZR9%h{s`e?6Fnud_q1tz~;?D}cJ=iFU-ShhJnJK^61@EN2CfWU3Jt
zGsT@`-#g3})LIf{fs!;%Tg~=97eToMQl#41nfy{^j>?DIk=s94Q75X*&;YF$DSlbW
zZsAhMnU{OW<zCjPym$|pTeXMCrsr1ddr%JcI%o2frr?RqVoDQv3Bunbl3j~zP;}5;
zIOpmEXKXKkxl~1coJXi7R<@+LTgO0;H*O@$v?F=+WE^$>;99oGzbOi_SPtKprNM%N
z@lax$jBZOEk?-witcB)4#p(yMDf>)IR5T`#wAge9N+V|wrCAX)_rA)OEpLO&sIrQ}
z`w@_%aw2YhM?s=ZFhps+(AA1=#QF#eSq}RErrN>J)$tIK^h4mtoF#$ZF7+TcpB0kX
zn=m=4okJ|U3}Jn$BT<)b+sOxrB;D5^2j4?$NJz(pBzyrOUTP^gt#gDmtE$N6&`P2(
zm2f8Q&C%i>XV{0COCjy_CbGb6Fp<t?iF$n{+*-F1%KzL7)<1lJkBp2olV-4%auej7
zpH6OHZ%O29Z&CLOc9FG1)M)9@rEq^wJYgi}fmIhqkl3W5WX_~C(sFwqIb87sHt=hJ
z!yzBy<QPr$Yi)~mKQ92A%M$2p>CcwUD5vZr&Jop;eBx{O2xRwTp;hR6u=LCZw<Esf
z?Hy}4HE1RT@oU(KrZ-^wkoj<}RU0yQ#aVK7&G+Q6<4@#Z00*+?+eq4w(<G?sB<t}m
z8JgslQLY*5z`>yyWGT-fUPj2+LuT;tblboI>&w8?YXj9Sw_k;w?*&l)C`G5XWkGU`
zElMwmqPjnHhR!Z!q*=#R@S*u`kaqqNs7H-Z*UBm=yZ$ZMmPD~B<vmE`l}A)zUNH>c
zEhj7fcu93Hs(|YUSHR$pdUjgGO{lD~hRyG%g87#B@W<50fb6fpg@At%;(wH!+i!th
zO*;vuls&o<djsIXR`_tsmx_Hc0NvkLOq~4V;Pr1VX!)t#fKQzwZt35V*5nS@m~4Y}
z-#H@F0hi(N+0N*Fpg(%EcO{gSJAs$$J`x(Uk@9(2MieUtAm6F$As}f%px-hxl<g8g
zrX-jmljD`pVeB&4RBnx;S~q3a{PuGH;)9{2N%}wpzGtYnA7ju>>;WHRM}sTc0k^v@
zrPf~E3R%UuB(nAa=(m;-yx|D6v^7UNf0zz6`ArD!*q)k>nxm`sF+_D%25nXc!OiOC
z@S;G%b#D5aCCpB?)SM$JReOoW#Z_$Zuh$`W@<ysJ#DnXJWb)mXnJ_sr56%pm1pc!n
zT*T%A(({}%ioG<6`uDp^=+dYul8k%|Z9{V4*PRT^o9c{g-z1Rh7NxKyX&vcrISNi}
z$|3E7kHPMNn_$PySn_W1F9Zd}k$}^qN%I_vtnQu6X0^2?V|VQ&hkmMoCV!rQJ;%ez
zkqhVHn{_f&UYi2ZhtIHU-L&ASA46*5E<mYsInnvMgZ^S3Y1E+yRJ#;b2s``k#O0ym
z_@kT9{%L3LT4!hX3JdA@H9NaJBotakRf8cv0BsK43oG$lG<o`2=rX4l3Y#rKLjDeg
zl$$I6Uxm=VHox~*=8vp2XSs+1^wu^Bq2GTZp)buj&pWHxintAA#x*ar&G9{1dMJvU
z_Iv@FgKOdHnsVrK^ezdQZ;z(DcnSFvmO{&o9f)mw56*6!Bg)wQBU=CA8oM~ii(S1w
zkb697H|+VPCE44#FPG^%7OC1h23{#yg(S=8Ao-%XXfU=#UXnSeTlZ3OKlc!M{lJOK
za_~YK2RflK=Q(h+nuG|OgZ34apfSps+`QIZkSuWnS(Mcl<#~m3zb_lW&HAa9P0X}L
zXB>8s{H{yU4Tsn4x;CfaSbzuGI;{hD;7xl7`E46qn>&n{xPM?*oL@k8?J9(C3wn#=
zn|^Ghg+w@Z^X;F;*Rsz1%ECg-g!v5~RR*UI_@Rg4R_KB48usyyxnQrhMM~G!{Qn3!
zlrVpi)mv{0b0@Ta;;&mjK3_S1bJAl`*JHCdU#Irmla}S&6!}qV?!*X`{b2#uJa-<+
z@3@4EU*UmfY(L8lS=t;Wy}DR&q;@2icXu7}Sk{hWA8dvZ>;&%Dfit+EzNUz2Yk{Ql
znOwrgY+{<Xo@z3*kZZSiI$ZrzgKUmgaBV6^a!xsG;jL{cM1JGR{TTibZc)`Bi@3+V
zO9<x1YUY!Pah1^csXJP^s}(xbY5|w{#|Cc4G6k&gIY+%8;RsN42$DB#AvLR{-1et3
zZpx11)YBoZNcZsq+&DLt8&JEBTl--lwdT}XB8;E<vh2-me(Z$ywd3demE)(q+7!Ic
zdy?)aQpnata|7FTJ#)~>)*6^W-+_?Z?{3Cc7e24h{yuk8$v4K@_;f807SiQwe*Ap>
z$gG4c=Ss&96)q8k_V-zwCBDWi?&7_cf!3=P8ues_a7h)fR)&RmePz-*rwGqpp1u8i
zddXyde*T_*z5IInddob${k(g5%6j{G`u6ts_3rKI-OEo<d35tYxf`!gXwh;LLF|{C
zx4P#oYvGG}PrZA%hS0hPC!2b(PW)L)gnl`mHN@6knuR;Ozn=WhSdqZ9A+YY6S|6Cg
zYxttq!>b{%?qX3N_^ck#yCJadDo`I7-K=g#Swmpm@~;ovQ4i?T5LmZ%>jR(J)y?SJ
z5LmYyoBlPVyrpUW3HYhd8zSqLh;u#U`OnDzUA)n93yU6~0By~Bet`-%V#4$JGg_=V
jiDZ(!(a+<AOI1l#3*)=6j7Q7Og+n-xv62Y!#@PP=sjg?e

literal 0
HcmV?d00001

diff --git a/AI_CHATBOT/intents.json b/AI_CHATBOT/intents.json
new file mode 100644
index 0000000..ccc4ae7
--- /dev/null
+++ b/AI_CHATBOT/intents.json
@@ -0,0 +1,138 @@
+{
+  "intents": [
+    {
+      "tag": "greeting",
+      "patterns": [
+        "Hi",
+        "Hello",
+        "Hey",
+        "How are you",
+        "Is anyone there?"
+      ],
+      "responses": [
+        "Hello! Welcome to PETARI, The Food Donate WebApp. How can I assist you today?",
+        "Hi there! How can I help you with PETARI today?",
+        "Hello! How can we make your experience better with PETARI?"
+      ]
+    },
+    {
+      "tag": "goodbye",
+      "patterns": ["Bye", "Goodbye", "See you later"],
+      "responses": [
+        "Goodbye! Thank you for visiting PETARI.",
+        "See you later! Have a great day!",
+        "Bye! Come back again soon to PETARI."
+      ]
+    },
+    {
+      "tag": "thanks",
+      "patterns": ["Thanks", "Thank you", "That's helpful", "Thank you so much"],
+      "responses": [
+        "You're welcome! Happy to help.",
+        "Anytime! We're here to assist you.",
+        "My pleasure! If you have more questions, feel free to ask."
+      ]
+    },
+    {
+      "tag": "mission",
+      "patterns": [
+        "What is PETARI?",
+        "Tell me about PETARI",
+        "What does PETARI do?",
+        "What is the mission of PETARI?"
+      ],
+      "responses": [
+        "PETARI, The Food Donate WebApp, is an initiative aiming to redistribute excess food from various events to those in need. Our mission is to achieve a hunger-free world and zero food waste.",
+        "PETARI focuses on collecting excess food from weddings, parties, and events to redistribute it to individuals who are hungry, thereby reducing food waste and ensuring that surplus food benefits those in need."
+      ]
+    },
+    {
+      "tag": "features",
+      "patterns": [
+        "What features does PETARI offer?",
+        "Tell me about the features of PETARI",
+        "What can PETARI do?"
+      ],
+      "responses": [
+        "PETARI offers several features, including redistribution of excess food from events, alignment with India's Sustainable Development Goals, environmental consciousness by reducing food waste, and societal impact by addressing food insecurity and hunger issues."
+      ]
+    },
+    {
+      "tag": "sustainable goals",
+      "patterns": [
+        "How does PETARI align with sustainable goals?",
+        "What sustainable goals does PETARI support?",
+        "Tell me about PETARI's sustainable goals"
+      ],
+      "responses": [
+        "PETARI's mission aligns with India's Sustainable Development Goals, particularly focusing on making the world hunger-free and reducing food waste to contribute to a sustainable and equitable society."
+      ]
+    },
+    {
+      "tag": "environmental impact",
+      "patterns": [
+        "What is PETARI's environmental impact?",
+        "How does PETARI help the environment?",
+        "Tell me about PETARI's environmental efforts"
+      ],
+      "responses": [
+        "By redistributing excess food and reducing food waste, PETARI helps alleviate the pressure on finite natural resources and minimizes the environmental impact associated with food wastage."
+      ]
+    },
+    {
+      "tag": "societal impact",
+      "patterns": [
+        "How does PETARI benefit society?",
+        "What is PETARI's societal impact?",
+        "Tell me about PETARI's societal contributions"
+      ],
+      "responses": [
+        "PETARI aims to bring benefits across society by providing access to food for individuals who may not have adequate means to access it, addressing food insecurity and hunger issues."
+      ]
+    },
+    {
+      "tag": "economic impact",
+      "patterns": [
+        "What is PETARI's economic impact?",
+        "How does PETARI contribute to economic sustainability?",
+        "Tell me about PETARI's economic contributions"
+      ],
+      "responses": [
+        "Through its activities, PETARI not only addresses social issues related to hunger but also contributes to economic sustainability by efficiently utilizing excess resources and reducing wastage."
+      ]
+    },
+    {
+      "tag": "collaboration",
+      "patterns": [
+        "Does PETARI collaborate with events?",
+        "How does PETARI collect food?",
+        "Tell me about PETARI's collaborations"
+      ],
+      "responses": [
+        "PETARI collaborates with various events, such as weddings and parties, to collect surplus food. We emphasize the importance of partnerships and community involvement in achieving our goals."
+      ]
+    },
+    {
+      "tag": "volunteer",
+      "patterns": [
+        "How can I volunteer for PETARI?",
+        "Can I volunteer?",
+        "Tell me about volunteering opportunities"
+      ],
+      "responses": [
+        "Thank you for your interest in volunteering! Please visit our website and fill out the volunteer form to get started. We appreciate your support in making a difference."
+      ]
+    },
+    {
+      "tag": "donate",
+      "patterns": [
+        "How can I donate to PETARI?",
+        "Can I donate?",
+        "Tell me about donation options"
+      ],
+      "responses": [
+        "Thank you for considering a donation! You can donate through our website by clicking on the 'Donate' button. Your contributions help us continue our mission to reduce hunger and food waste."
+      ]
+    }
+  ]
+}
diff --git a/AI_CHATBOT/model.py b/AI_CHATBOT/model.py
new file mode 100644
index 0000000..adf38dd
--- /dev/null
+++ b/AI_CHATBOT/model.py
@@ -0,0 +1,19 @@
+import torch
+import torch.nn as nn
+
+
+class NeuralNet(nn.Module):
+    def __init__(self, input_size, hidden_size, num_classes):
+        super(NeuralNet, self).__init__()
+        self.l1 = nn.Linear(input_size, hidden_size) 
+        self.l2 = nn.Linear(hidden_size, hidden_size) 
+        self.l3 = nn.Linear(hidden_size, num_classes)
+        self.relu = nn.ReLU()
+    
+    def forward(self, x):
+        out = self.l1(x)
+        out = self.relu(out)
+        out = self.l2(out)
+        out = self.relu(out)
+        out = self.l3(out)
+        return out
\ No newline at end of file
diff --git a/AI_CHATBOT/nitkutils.py b/AI_CHATBOT/nitkutils.py
new file mode 100644
index 0000000..e69de29
diff --git a/AI_CHATBOT/nltk_utils.py b/AI_CHATBOT/nltk_utils.py
new file mode 100644
index 0000000..690b8fd
--- /dev/null
+++ b/AI_CHATBOT/nltk_utils.py
@@ -0,0 +1,23 @@
+import numpy as np
+import nltk
+nltk.download('punkt')
+from nltk.stem.porter import PorterStemmer
+stemmer = PorterStemmer()
+
+def tokenize(sentence):
+    return nltk.word_tokenize(sentence)
+
+
+def stem(word):
+    return stemmer.stem(word.lower())
+
+
+def bag_of_words(tokenized_sentence, words):
+    
+    sentence_words = [stem(word) for word in tokenized_sentence]
+    bag = np.zeros(len(words), dtype=np.float32)
+    for idx, w in enumerate(words):
+        if w in sentence_words: 
+            bag[idx] = 1
+
+    return bag
\ No newline at end of file
diff --git a/AI_CHATBOT/train.py b/AI_CHATBOT/train.py
new file mode 100644
index 0000000..c2e0e04
--- /dev/null
+++ b/AI_CHATBOT/train.py
@@ -0,0 +1,108 @@
+import numpy as np
+import random
+import json
+
+import torch
+import torch.nn as nn
+from torch.utils.data import Dataset, DataLoader
+
+from nltk_utils import bag_of_words, tokenize, stem
+from model import NeuralNet
+
+with open('intents.json', 'r') as f:
+    intents = json.load(f)
+
+all_words = []
+tags = []
+xy = []
+for intent in intents['intents']:
+    tag = intent['tag']
+    tags.append(tag)
+    for pattern in intent['patterns']:
+        w = tokenize(pattern)
+        all_words.extend(w)
+        xy.append((w, tag))
+
+ignore_words = ['?', '.', '!']
+all_words = [stem(w) for w in all_words if w not in ignore_words]
+all_words = sorted(set(all_words))
+tags = sorted(set(tags))
+
+print(len(xy), "patterns")
+print(len(tags), "tags:", tags)
+print(len(all_words), "unique stemmed words:", all_words)
+
+X_train = []
+y_train = []
+for (pattern_sentence, tag) in xy:
+    bag = bag_of_words(pattern_sentence, all_words)
+    X_train.append(bag)
+    label = tags.index(tag)
+    y_train.append(label)
+
+X_train = np.array(X_train)
+y_train = np.array(y_train)
+
+num_epochs = 1000
+batch_size = 8
+learning_rate = 0.01
+input_size = len(X_train[0])
+hidden_size = 8
+output_size = len(tags)
+print(input_size, output_size)
+
+class ChatDataset(Dataset):
+
+    def __init__(self):
+        self.n_samples = len(X_train)
+        self.x_data = X_train
+        self.y_data = y_train
+
+    def __getitem__(self, index):
+        return self.x_data[index], self.y_data[index]
+    def __len__(self):
+        return self.n_samples
+
+dataset = ChatDataset()
+train_loader = DataLoader(dataset=dataset,
+                          batch_size=batch_size,
+                          shuffle=True,
+                          num_workers=0)
+
+device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+
+model = NeuralNet(input_size, hidden_size, output_size).to(device)
+
+criterion = nn.CrossEntropyLoss()
+optimizer = torch.optim.Adam(model.parameters(), lr=learning_rate)
+
+for epoch in range(num_epochs):
+    for (words, labels) in train_loader:
+        words = words.to(device)
+        labels = labels.to(dtype=torch.long).to(device)
+        outputs = model(words)
+
+        loss = criterion(outputs, labels)
+        optimizer.zero_grad()
+        loss.backward()
+        optimizer.step()
+        
+    if (epoch+1) % 100 == 0:
+        print (f'Epoch [{epoch+1}/{num_epochs}], Loss: {loss.item():.4f}')
+
+
+print(f'final loss: {loss.item():.4f}')
+
+data = {
+"model_state": model.state_dict(),
+"input_size": input_size,
+"hidden_size": hidden_size,
+"output_size": output_size,
+"all_words": all_words,
+"tags": tags
+}
+
+FILE = "data.pth"
+torch.save(data, FILE)
+
+print(f'training complete. file saved to {FILE}')