|
方法ConstantMean(),num tasks=237)39 s e l f。covar modu l e=gpy t orc h。k e r n e l s。M u l ti t as k k e r n e l(g pyt o rch.k e r n e l s c a l e k e r n e l l(40 gpy t orc h.k e r n e l s.MaternKernel(nu=2.5)),num tasks=2,rank=141)43 d e f for w a rd(s e l f,x):44 mean x=s e l f。平均模块(x)45 c o v a r x=s e l f。cov a r mo d ule(x)46 r e t u r n gpyto r ch。d i s t r i b u t i o n s。M u lt i t a s k M ul t i v a r i at e N o rm a l(平均x,c o v a r x)48 t e s t x=t o r c h。l i n s p a c e(0,1.0,t e s t in g n u mb e r)49 t e s t y 1=t o r c h。Float T en s or(c a l l(np.a r ay(T e s T x)))50 T e s T y 2=T o r c h。Float T en s or(put(np.ar ray(T e s T x)))51 T e s T y=T o r c h。s t a c k([t e s t y 1,t e s t y 2],-1) 53 l i k e l i h o d=gpy t orc h。我相信我会的。M u l t i t a s k a u s i a n l i k e l i h o d(num tasks=2)54 model=多任务GPmodel(t r a i n x,t r a i n y,l i k e l h o d)56 model。t r a i n()57 l i k l i h o d。t r a i n()59#使用adam o p t i m i z e r 60 o p t i m i z e r=t或c h。optim公司。Adam([61{“params”:model.pa rame te r s()},#I n c l u d e s G a u s ia n l ike l h o od param m ete rs62],l r=0.1)64#“l o s s”f r GPs---该m ar g in al log l i k e l i h o o d65 ml l=g pyt orch。是的。Ex a ct Ma rgi na lL og L i k el ih ood(L i k e L i h o d,型号)67 n i t e r=30068 f r i n r i n r(n i t e r):69 o p t i m i z e r。z e r o g r a d()70输出=型号(t r a i n x)71 l o s=-mll(输出,t r a i n y)72 l o s。向后()73 p r i n t(\'i t e r%d/%d---Lo s s:%。3 f l e n g t h s c a l e:%。3 f“%”(i+1,n i t e r,l o s.item(),模型。cov a r mo d ule。da t a c OVA r m od u l e。
|