From 99cfcfb47ecb756b8aa43643871fee24105b17a3 Mon Sep 17 00:00:00 2001 From: ym13n22 <ym13n22@soton.ac.uk> Date: Tue, 23 Jul 2024 17:43:26 +0100 Subject: [PATCH] train data --- Windows/mlp_model.py | 18 +++++++++++++++++- Windows/trained_mlp.pkl | Bin 6847 -> 6847 bytes 2 files changed, 17 insertions(+), 1 deletion(-) diff --git a/Windows/mlp_model.py b/Windows/mlp_model.py index d93feec..9401cea 100644 --- a/Windows/mlp_model.py +++ b/Windows/mlp_model.py @@ -108,11 +108,27 @@ class MLPClassifier: self.train() self.save_model() + def predict(self, X): + self.load_model() # Ensure the model parameters are loaded + with torch.no_grad(): + y_hat = self.net(X) + _, predicted = torch.max(y_hat, 1) + return predicted + + if __name__ == '__main__': mlp_classifier = MLPClassifier() file_path='trained_mlp.pkl' - mlp_classifier.run() + if not os.path.exists(file_path): + mlp_classifier.run() + test_data = torch.tensor([[97,58], [25,34], [42,7],[298,9],[10,39]], dtype=torch.float32) + + # 进行预测 + predictions = mlp_classifier.predict(test_data) + + # 输出预测结果 + print("Predictions:", predictions.numpy()) diff --git a/Windows/trained_mlp.pkl b/Windows/trained_mlp.pkl index 7719b6dbe8ef20f94b6c89ec71a8ed500902906a..7c0431e06e4de36b9fe4631f7b23032980e9dca0 100644 GIT binary patch delta 1665 zcmdmQy5DreT}ECb3ln2wV`CE&0~0fo$pt)enJ6NKjEQ<83=nW|L6`kR!+-nchTgH8 zwWPsL*yFA3dDgZ2T^Bd(*V5OucZ@o0AN_gr-q2m^_v|pU+V^r<;r@W5tL?d^%k5q6 zdF~IGSY}tYN`8M+-1U883f=oQ7cy}oJHX7;Y;v7|oFl>kW~OE!2XH~`XSgK8<^W<Y zbDeHK+k$!j)IxDP5MOcQBzyZ>&Hd8F{rgWpyS#7mT>;6-|Cui!dCuI_X!0Evxq5`z z=B7p<vo$~lFfcHPB&Y75YT|6)^yT!vBNN}*6}8vw59i-zFWOmeza@UH{oZ#w_FrSR z*gw-_-~Qf9;s+xS-rMK9YW4nz1@Zf37xdaO9oTMP>UqU3R8Vifkz2QoeD?Z%Gj|B? z6Bb=<J85Bn{WI&T1Lqqp=k1G;3bp4dj&v~Je%t=vzn=XSnn(9_hV|G_uwQF`Zt-#Z z3UvvGeVmu~etf;x{&?Xx`{NtG?khYKHu<2K+~f<aNPaRhFff?BmQ79*?kA84$WJU# zKe@j4+8>*|-~RN5_xs-6RdJZ?C<O`^?#&YH>MZcUF*P+bHJZFZTn<g7mtT4EEbfyC zQ3GRBBa{?k3K9WX2umT$Zhp50*?qZA)q4J&r@I>LVl6@Z?n`yMB{-CK7g}E14R+kc z0t=A1#NIm=P4zsx<4!FB>+{L#*u#~_X8rGh@E(x5$%RZJleu}XBHUnVWMTnwLk7&# zrXUeyH*}|b0VfjW<${jR-&Q-?uyup^+ND00ugub%Kh2YHyeIJ9DeX`QIPp|usoUlY ztg*}M{SW5%Wvg4u*)ux`NdB{P4Ou+7fJt;RH!tVpiF`+R5SE%7O-|&OM-}Mde~NIZ zp^+hKYBmLlfE)@-&2QoxcY^|9$(lR6^!~50)TmGf2Ubbw=ACh;HLcIB)v*Q#!d-z| z2y=}LEDTU`i77|~+1$LcTUH=*Beuv`?Yv@Tz2(t*F#qCz0W+@DwKlhwKebY5HMa)m z5*wi-2(t|hjSNr|0$8M%Uwg6`%T9!6jVw$i-)51AXC_l43lj^F6JVLCVULLAhP}#G z3>SkeKJbL?0yzg6ciq~vOGj(|t{i4P>;8jt4JCedS^Z90WX;S!ZMT8@o?RCE?pk?W z6}FzgRb+R}pW7Jve=i6%pD1W`NI~_{Het5~q1(M|+js4ea<hK%_Sr5o<AR<0CLi7X zXd#QuI=^kZ|2d|D!;yu7p#h2+CW+odc-z#{$N;5qFa?Pq`$xQRhvW3;%N@i%tZ_V+ z{sWv=k^RGH#Ot_TkK4ZD<|~IyztYuC9hG!U^JuaUxc}SX$lZL~jp93PAH^)R)jh^! zr~HM}23aqNeQ<T0;cIiNP>x1M28IKx;|%|D?{j3E{LnsZ^>oLB*G^cTo#pAYPeROb z%2ZRQ^vFlnftTjk>qmdFG0)7hXTHg2HAz%<@_jL+s5CY*M$6M+5l~b<XWfbLyP>h6 z5lSIq3K9YN9hCGyZoBEe&r0Mi<8B@EOIEj*^y~r`B07=>5oQ>H3U1_r$P^@kY=+@3 XDQj`gRaPfMSgd1K%UVo!l&S*&=T`{y delta 1663 zcmdmQy5DreT}B>5GeZkAQ*(1OOAGVK`8@KO$O1-%jEQ<83=r_i=9evt(0Y5l19kR7 zO04^@%y+W?w&(7?{le$%-$>XVh|((D-)h8f7vQwtuEjWi*Fw`W`w7=L_j!EhvUf<9 z+J9P1Z-4)xdi(u?Z|$BxyKJ|)kcpF#7iOKgfuWhf<T?R4M-&l|1GpgeGc0hMwr|GH zW&6EOD((jfZRU7j52CB3bnH74<oAckl-Y^PFSnh1S3q*|f94Btv(3y6EG#D9VUeqc znGF^JnXLgbfPsM_N&b`l=lweN7gaOuC*S{X|LRqsJx9<i8`deC?Y|bN+0TxBuy2hQ z-vOTdSNmC6`}WU$qhp`NFTL;6j5+oP^bgs;-`BVQ-Ig=<kNT|kFSHllKkN6>{ozT) z`xehEv-=t8ZFh~SY+u`5tIKv~^+C2%XRqHE`K#2{-%iEeT>9kRcBY1X7mm)|_x*c= zoxEf5{%+@fJLg5-`_E_v??0*4VK<*Yz;2SL?Bolq2tSz{SQ=SOUdtw@3G)+J1mq_c zsGs(23D|!n^{Bmj$bo&XUmEQvJ4%7Vg?qCEyE+Rz;>}HsOeQDt%cBbP@T*Lo#eEVU zgyyCuCI*uWc;w*e11tiv4wgQCM4h%;qpiA!fkDLt<bo`@BpVPtiKD<OWA`RYoe4@- zUi{B?&&cT8trWqvdsgMX-5~j=J{(|mGZtO4DtWrrD&)u#%L7h@J0}-1iA?6^y^7=p zQ)4qUH<%imA-kdVU8d8-=M0Wpy_PzG%=U_0<_Mx)JdWF&f8SyI?Eg3Ge;G?08T9k* zrLu3>z2VslPC{Yn?O<B}l!)zH&360n*?VjcvHIFW-M~3{BHs~209l%vp@tYppojk{ z!lmYBCKjlP86*O7DJ(IUFI%!RQ$ogCVzQn!$c1YqW`aYbrTO0WIgXVU-q-YYEfnS7 zJ^8M{Ex5TBh9(xM=7L3#&7CjZW-Yv5!kYL0eXzNKow;CI?2wFQ$;A0q&Ry3mw)}Lq z24@l*p(99Un_8Gm-XJcACeq8VJz0!pC&Kp@riPPmv&h5qlDUPcA-JG`<)vB2*>}HC z7P9v7U%qS7^hs7J|LS(<dCJ)Q`4YQFAcxWB*p;cfUp<zw3EwZYho`p@>^x8q!0`O~ zdDd4xwpt6ktJ$shb167jU~(W@>%&>=i=0;1t)*W^?(tvx*0QT4)ynIEsd=b?g*t~% z#;&SWEml=)xXjN;i0hr%-?}U1Q~Iu70W8*OA2N4w)Yk5tBzh0-Npo{^6BCrO0W1Ov zQkZ`ZwuU>@`m@?KJDNCj`fay!|Dxpdn)$WEjw#Mg(-O=aS-d2iZr|!~@OG1TvbZMz zPH-T1!mxK8t7D$BtV3IytK+WVMhBQUjK069)amzaD>cVRmtLnNrwqrRn)7WpntrvN zU#)Fzv7^F{Rer6F{N))otwN3F1(Ke&|LPm<%5NFkUAHr^ceryDTsGVnLyAfRLld;9 z1dD*8@;U2Hgx@WV&CO8D5ReGS@31mt*ZmZ0ZQI$qUZicc>ibf-8(fCyNFIcnVPR-s gfLacLMUc%{9;s_BA)#lL`B2+>ar!>1$&ONW0PnyG^Z)<= -- GitLab