我不知道这个问题是否正确。如果不是,我很抱歉。 我是fortran的新用户,并花了很多时间来处理以下事情。
我构建了一个名为“loglike”的函数,它根据两个参数返回一个实数。我想用这个函数来构造一个mcmc算法
就是这样。
psi = min(0, loglike(propalpha,propbeta) - loglike(currentalpha,currentbeta))
其中propalpha = currentalpha + noise
和propbeta = currentbeta + noise
,噪音是来自某些发行版的随机样本。
现在我想通过调用先前构造的函数“loglike”来使用这个算法。
1)如何为称为主程序的新程序调用函数'loglike' 2)如何将其用于子程序?
任何帮助对我来说都非常好。 提前致谢
编辑:
module mcmc
implicit none
contains
subroutine subru(A,B, alphaprop, betaprop)
real, intent(in)::A,B
real, intent(out)::alphaprop, betaprop
end subroutine subru
real function aloglike(A,B)
real:: A,B,U, aloglike
aloglike = U
end function aloglike
end module mcmc
program likelihood
use mcmc
implicit none
real :: alpha,beta,dist1,dist2,prob1,prob2
real:: psi,a(1000),b(1000), u1, u2,u, alphaprop, betaprop
real, dimension(1:1):: loglike
integer :: t,i,j,k,l
real, dimension(1:625):: x
real, dimension(1:625):: y
integer, dimension(1:625):: inftime
alpha = 0.5
beta = 2.0
open(10, file = 'epidemic.txt', form = 'formatted')
do l = 1,625
read(10,*,end = 200) x(l), y(l), inftime(l)
enddo
200 continue
loglike = 0.0
do t=1,10
do i=1,625
if(inftime(i)==0 .or. t < (inftime(i)-1)) then
dist1 = 0.0
do j = 1, 625
if(t >= inftime(j) .and. inftime(j)/=0)then
dist1 = dist1 + sqrt((x(i) - x(j))**2 + (y(i) - y(j))**2)**(-beta)
endif
enddo
prob1 = 1 - exp(-alpha * dist1)
loglike = loglike + log(1 - prob1)
endif
if(inftime(i) .eq. (t+1)) then
dist2 = 0.0
do k=1, 625
if(t >= inftime(k) .and. inftime(k)/=0) then
dist2 = dist2 + sqrt((x(i) - x(k))**2 + (y(i) - y(k))**2)**(-beta)
endif
enddo
prob2 = 1 - exp(-alpha * dist2)
loglike = loglike + log(prob2)
endif
enddo
enddo
do i = 2, 1000
a(1)= 0.0
b(1) = 0.0
call subru(a(i),b(i), alphaprop, betaprop)
call random_number(u1)
call random_number(u2)
alphaprop = a(i-1) + (u1*0.4)-0.2
betaprop= b(i-1) + (u2*0.4)-0.2
if(alphaprop> 0 .and. alphaprop < 0.2 .and. betaprop > 0 .and. betaprop < 0.2)then
psi = min(0.0,aloglike(alphaprop,betaprop)- aloglike(a(i-1),b(i-1)))
call random_number(u)
if(u < psi)then
a(i)= alphaprop
b(i) = betaprop
else
a(i) = a(i-1)
b(i) = b(i-1)
endif
endif
enddo
do j = 1, 1000
print *, A(j), A(j), LOGLIKE
enddo
end program
答案 0 :(得分:4)
最简单,最可靠的技术是将您的函数和子程序放在一个模块中,并从主程序中“使用”该模块。这可以在一个文件中完成。此方法使得过程(函数和子例程)的接口已知,以便编译器可以检查调用(实际参数)和调用(伪参数)中的参数之间的一致性。草图:
module mysubs
implicit none
contains
subroutine sub1 (xyz)
declarations
code
end subroutine sub1
function func2 (u)
declarations
code
func2 = ...
end func2
end module mysubs
program myprog
use mysubs
implicit none
declarations...
call sub1 (xyz)
q = func2 (z)
end program myprog
ADDED:“implicit none”用于禁用隐式输入,这在我看来很危险。因此,您需要键入所有变量,包括函数中的函数名称。您可以从模块的其他过程调用子例程和函数 - 它们将自动被知道。因此,如果您愿意,可以使用“sub1”中的“func2”。对于模块外部的实体,例如主程序,您必须“使用”模块。
答案 1 :(得分:2)
这是它看起来的一般方式。请注意,函数通过将结果分配给自己的名称来返回值。 1}}声明不是必需的。
return