From 3b9252d93deac66e76101be4741429bc24ee65e5 Mon Sep 17 00:00:00 2001
From: HaoYizhi <haoyizhi@agentlab.cn>
Date: Sat, 10 Jun 2023 21:26:35 +0800
Subject: [PATCH] cap prob type and level

---
 __pycache__/controller_db.cpython-38.pyc | Bin 7623 -> 7659 bytes
 __pycache__/firm.cpython-38.pyc          | Bin 7068 -> 7047 bytes
 __pycache__/model.cpython-38.pyc         | Bin 11842 -> 12201 bytes
 __pycache__/orm.cpython-38.pyc           | Bin 3822 -> 3883 bytes
 controller_db.py                         |   6 +++--
 model.py                                 |  28 +++++++++++++++++------
 orm.py                                   |   3 ++-
 xv_with_exp.csv                          |   8 +++----
 xv_without_exp.csv                       |   6 +++--
 9 files changed, 35 insertions(+), 16 deletions(-)

diff --git a/__pycache__/controller_db.cpython-38.pyc b/__pycache__/controller_db.cpython-38.pyc
index 4452c70b94dfccf17200fc953723ea5a086d7481..4042aa17a72714323cdc1e13c518d95818e7ac45 100644
GIT binary patch
delta 677
zcmZ{hO=uHA6vz9LG%dQ>wAmykn^?qtr6^Kzk`|%Ri0DlvAUSl|WT#|h<0iAKRY|pw
z3R>HXPr+{l5ih-H>dBK{DjxM9?nUrgdJqp@DmZUf>dArm&42!H-h0E$=JeLIT#m)|
z^oVhMVc94idL<VnH9`BoLlJ9A%vmv=b6Am%$?ZWSyrwKB-0-tVJAu!oF;MWMv_7iT
z^rmK+4b#<{+^%Zwl_vAz|4fTrViqpNPK)!OV&!qJi=rMdo)v2VmDDMMn9dP_o_;(j
zkM=@nJDTdYWCm~-zr@#}ipvADVBm*=nY=BIdEf*k&WjEA>kJ8Xr#mvZ5rR5?Rfc*k
zVZ6j0<pf;EQgRU*_&zxaOZYQ+HglEIsH+#P)oruJc!Zp5IF~9Z7fJT{!Fkhi8Rsp0
zlPbvzzGX0r3D3a3^ciSlC0&3Q-A;N7;4<176*{<*8HexqI<r}6P^C$M35oBv<TuE@
zNeB^k+wm3JQWst&v<Qa<Cc>3|%W*YFGY#G$_db5k7T^c&WZzBwH=k1ZLqeR2MEE|^
zK|U$OU+O{ljcRTR5X-rlu}5?+%2z46M)0q2o0NZ5gVYmzn;V9`_$_x!{EL|%F5Dyg
xfq*ADuI`$(26OAS5%hjcL3&O86gTom;Q(&uO(l5yrhdWD9bTm9Dw_M}{s8P~t91YX

delta 638
zcmZ{hPe>GT6vuncx~9yI&hG5G?u?5h*=}_ZPqCs13e~L=JM5uln3>waQP(#^A-V>O
zApThhU(sJ72&sbvv7Neg^3pkKFLvr;<gtSU{oW|v`oZt>-tYU~`~4p8cQCv+tW3pX
z9cRQU%&*v=E`Csk<c!MalNA?M6MI~2|27rn5!sgN@v0g~uOmw;?Z^WVr!VquN4sTK
zN}fC82DBZ!Da6Cr6kS)0=w-gDji#0J((kC&c%nD}TXfX93nsnm8V8Ghb&VObBB=>J
z_6UnEBg4oFJZfB1KeoU<`kTnMP77g&`n0RCOl7SK9{tn?;URS=C({dDF)z@JTID&{
zaxlV5m1@a}M8Ixa7`*FxfrD6~@5u?}ekj?F=LpTBOQ~D1NXw}lylK2m9f0vb#PhA#
z&1n)kK28b`Z!l`~K79dx(AV@{(PLpyATQ%-AGpHmDx-yQ+5@ifk)KABQDN|)@>21v
z=LaRf<l0zc^(p=7$-yD@Wj+`FThFEVoYBdbM8bAY>RH_)T|Wm$G^!WG>=*R0fft+?
z#SM;bGQt+t*$Z2>*n36$dOx(2WUPtLsvG^e$1Fb+Xv%(I2Cg;Z1m!vV`0bY*<PqaG
ceKW2=megK1ar_|_bJjL}9OCE()qC&!1Ii?wIRF3v

diff --git a/__pycache__/firm.cpython-38.pyc b/__pycache__/firm.cpython-38.pyc
index dccb9943c35f79f7a2ffd0d657ddf8bf3a6feed6..9e75445f1093a6bfe6f8cedeb25d1b93e7819bdd 100644
GIT binary patch
delta 328
zcmbPZ-fqqp%FD~e00f-HEh(4vHu9AV2u)*TVCZ1XVpzbqkimsv0%L5;<mm!Ic0eK4
zq7OApE)20PwJbm$TNZm3M;0fP#Z|*1$&kXF!qUrB%L<p}hRIGg7PMF7sbSA%E6S>2
zUcg(!vXGIHp)jk2xrVu!F@-UkX#!)B)8xs5>a1*8d|CXH_X~PMS^PpX7*|eSA+(%v
z&E$vNij(VjI466H*fVaP+%1yIxNWk6NEYKh5S#xX&?ChxK!S~t>A$AH<o_aSlXXOw
zi<g0X$RxtZ$H>FT0>na$JWMQ%Y@6SRo@Hb_IeC}(Z^jdow@EBxd^|Z;at-6N$y`!X
n7++1EFO?*k0MuC&4<eF5L<)#Vot!5vK3P{<oY8Eut8_R3f8bYm

delta 333
zcmZoSpJUD!%FD~e00dHnEh(J(8~MrwgcdO}Fmy0xF)UzQ$l$^-fiZT%<mm!IUL~wR
zLAGW_Mvx@i1cstNHB2rHu`RVMKt6jGM;2!m7nH?a!y?I$!kog=%Tx=}1(D@}$xc=m
zv{&S<Vb5kODyv~$z*obv5NK*)SqXCub2DQKV>Z(S#v;GT?Sks8>{<L-0+TljdP7;v
zLNgdwO`apPoN?{shun&j>v%XP+l$yUZkb#!lFGP!asaQ-<OC5p#{D2Z{~@5qidldJ
z8za+yO~J{ZMby}Cu@)3%=1rCmUBOr}`JkvMGZ!Pr=If$o85vJaUMl{Z@#N%%63ZB$
uO!kmm!}xsiSIH@iuP1j&C5a{ijVX!;5h)-d6-1;>W|Y=rG@ooB9S#7E&s{?R

diff --git a/__pycache__/model.cpython-38.pyc b/__pycache__/model.cpython-38.pyc
index b3619500fb1e5b7c5b98c37328573fe8cc6ccc50..04d2ad7c8b9fba63bdc6d693e89eeea193a3264a 100644
GIT binary patch
delta 3796
zcmZ`+eP~<P6@T|hdXi;*_%pT~EB>}D+ez$vIA2clWj9Nk+F6>Ui4sSVbDv~AJF<QC
z+<aY4rB17-Qm2LsrF6@98ig$zG^}Qf!RS8N*cc4*N1;rN(*3cqjt&CbV3f7lIZvwX
zq^ZUG>E3hBJ@=g7J?B38*B9<aeW$%%w*-I7*Kg5^*pGdK#J?G)9(Zi<q+hNOW8_M<
z-l*0jZNnlls-q2hl{IU$W`Xjo$(n5}&zh}SvmP`z=`CifW#3jH1(G3pyV+q>b5fAO
zSEt@(c2gPTGMA|>AekX*FAy9LOI+4^sDmohnO%|zPQz59(kq~(E?Pm|342@)NEur~
z%GlRwE71luW^X|nktW(N$QegQ;dZl+D_o(TY0zo?=H`M_AVgXw^WEIeZL8kBaTzwZ
zX<J|)*c|rGN|4Fcf|QpG$s8y}ic6DH#%V~}AZ*f@aiI(Df^D}IB<|#{VRZc(tlVzE
zyZ|YzW?R84zRV3%1Njc>1KF1euBEWEASFo_2y%1yP+veQ#cvK3B$R2psE>LePXCe(
zIC5*dX%+R->hH+wrL=05S~sm4E~k^J<du+)Tk9_;<H2z`8TYzm4tT6=<oPq04X@-L
zE1RG>XaI#Hkmm%;yO9aQ4FP%sKs(4H`6_)6C+1%9cUv3TEZp|}2_QY=&q;AY>uK#R
z7x#}yizHLEx)#>!`*_tN*(hbIc{Nl{O)ZS{VP388=iWt92lmu4ZC|#r$w#FrNI6pj
zD_q)B;$3@J!vS80iRIk6s<=!8)_#@Rp!kP9lpO@hnwB7T^ae##4^$lF^}HT4Tei*r
zd|94<7oNA_c@=8-5M+KBvn@amXe4^1RL3B}Iv(MbfTN)4)SlK>eNPNP>21gp{U}uY
z2JIULp->;RdC8W|`_b4uX803?Z%9Bpz_#}J9pLsE1Jmx*p0!*bhw;dIr$B6H+3j=d
z-GcLqfCLB>!kXjeSdIYAK};Lx+Bha#Aav7`lF2sk2JHkK^>3^|8feS3vTRR(s(+KV
zE|Qlq#|GLqEiXIRL9jdt^8lPkBYeBe=XoP-Z<G)lApcAN&LLo(iUjuk)R9t3l1zXH
zVD%}(KP;urSUx+zr=5L}{fg7m;10^_F=yA-JJZj=suy4@;A+h=u=WAC9x;#chDW$G
zR=FH0b9ojTwh3&#X7-$QHH=)4^Gdz~L}$%0L~Yzqrr|!qn#emqf;3hDjh&Cv$eRoY
zbhu;wK8&iPUJD}$pgwxu!VA_$rG1`;eH@vMd?PSE1dRK$hC@36lq=~OpuEfQYvbRR
zXg5gV`;Of@0*5wX_8ETbj#B1FmX-P4Yc7}Ni3e*=VfInppL68=Q0Sp@sH1CzST1tB
zkS~IdQJ!rEs)KrTQOY~@OFU>ooQu#66DOgxHo?=9Y2}-&K2zAoxI`P|Emohjf>lqZ
ztt@K;S-ashW4s+N89?a%I4}t1?3{-L=Rel>7|unKcjlbHV^1l#ER@kx`lP-eRf|L)
zE{(uxGd6fZwyuQ^-T@bhJRw=f<jW~JALJMQ+Wb1pL3P66d0`BPn)AY@z2y~W%N^hd
zWTi}(zDEygmvISi=AE=}$p$x3H<U;SW#(SqiFez?5A)FBn7;w9jl2!USQ9pI+h2Bj
zeATU#w}T$HMg8WlY%67YKu@)^SunAK$(FJSh)PZ}VSQgt2F+H~@NSd>Kl^#NbtU%l
zUKocL3GdNP)=AKkeY}r{40vM!-4AqCkMj^7CLN%Iw<NgawpnR~K$@^h{yi(5?341@
z{x$hPaF`JUEYlrN2-ph9sudS|AM%5v<zaYb^}`shF7tr(J;nR2i)log;#~*f8r*5v
zwecj0+u-(ta^d}OQP>?4zTtMmua9BLDj~M)ntva1V^B(O!SfoGo|nQyD?8jb2pJdm
zD({g~BI=nV=f!=`5=n}S-rHnW{L}k3NejcbNZt@0|Do})Ty!p^lJQw~rKqF;V>36l
zT#n5}XOdTvdUTG>PDb_Xb873XKBcl&eJYk}E!+3Emh9Oe{37~924LI7xPOw|6Yu%o
zA-@$bR4tGX#UHC)Cm)GO^$G7zFu5zUI<y=TKdtT&(=}f4>0_E*;#f^P`8UKK6#kl?
zj$t&%rEEXKVT4A6074VMf5Em}JaGGjt9CEhFOJl<lg~uFHX}<!T=F!E&ud33okWuU
z1;)~ip!ii~gP?V%S}d=J(9ILPdU0(tgis;=SGRo}!wB2h09ubA^dewLVVNDl@hHNh
zkyXK_Y5XCn*yYdN#Pv{?Su0G#$(3)_yU4ksEv>Ui?GyNpG=L`hB!z!XIHSI(&O~bC
zGx|D9OSocg*eD<N9ZIKFrYC1pqm0cmwiUK6x{@hfO{gp#siv2$6Q;O2b~&kEXWP)i
zF~`7$^}$&Dxsl0iL|h6a<?!Z}KLvuM$){e^S<Kq;iW*BrDpu#>KTX|aw`kt@lk8Dg
zypEP3obE-~gV2ny1EC$E9wChI6v94)T?icr4G4JHMMqlI7v{V$Om}0w5jXLW8sd}P
zvX89$*y_Ys^UsExhzuV)QLImutbCz-rnu2oerT6HvKx-WLlhV{!FOryBTeAbC$IDc
zHxO}T)0@yO#ZAqj7cq!S07as+(>S?`uz+w1VG3bP1Y1I+XJxd-XX~=61Sz+w<Sr?t
z^k{+H6xZ61DV9H7{I0z(a1-s%Afy3`&h*qmJU*kcDN)%mOBTe<j)D41sLmoxAfynu
z_+>{&+c_MbM8KY8Ndycd>@Kl5BwU?y@S_sM7NPdEuFegu7@cmJ+!mrc()t|eigJ<`
z9hYYppa`ribzET8`z7HGwFHrjq7qNe=qh8kQTYbKGD1!q3k{GlaU(QP{|egKpvu?~
zs;u7GC4Ld=CU;i;7W#saXT@K`JIH;})JtWyUA)vAbRPrD520<?GvfQbukL&a7hXZI
zh+AXtyC5rKdnqtbi07dKjra-vgT_oT**8kw5kKhLO(NnieXT>&D4qd`)FxCtnwq6*
zG^Xn;Ik}*#X_i5adv-yOQZ>O;wM1@Q?CS3(y13YXbj$ZF3za6+=uB)<ohe0g2i;mk
zdMJxsjiyxn>MWc7z4$}FQGW}SIfNyIw-A0L5}W%|L6^^!aP={aJX{lT#u92upSy0I
zW>J}orD!b8E`qY?NG)6e#W;*3uCn)QEP}2qMzIM>WrhEhMr?3@Gds#02yY^s7i7z!
F{{uK1<?8?d

delta 3384
zcmZ`+TWlN072R2qON!!CZ&IS(lt{fSIkqg@N^QyVE0*6*oHAlqSwqrXNu;z|!A?2^
z(NHOusO6}bHYjQbS)wii7e$L!C<>(cXoLJDaEgBPBOuVA$VUMO{Yd*!v;}JS-W4rN
zK_X%g@7$UDoHMi3{n-06uIHUjy8wUpGHdi`^jEGC;_io`10EARE1%i;IkK3m(!8=D
z*XN1mjWwtqW7KGjnkploG4iQ?y;*J11ICKh5-E`^QQP!(&C5th1nUm9Q}3c8=tU+{
zlTXmQskuZ@9v7G>2dRZh)VeMb9Q9Cv3a^VfYN9q;MeRv*Li7n)Q&Py9pH9<iT!b@v
zZ%LREqMb8h){>Q&Sr0LZNz^e9b8=YkD+wh+gblLX#LUd}#65K;L<nw@`yoyxI21Ba
z5{iN*=z}b0WQwLKA#2qHc?g0uWo_t!*&ytmlEAFYHjb_tm=0?&E<p|3)mE{powb2}
zgt|aisc=ULdrLwJ`(|hGp)Q|59XZTjf2Jg$Qr<^h)B$<A*G<qQyS$%zsFQl{iaWK`
zv)x)d^^6N!^<=AAHPmC5!<AwjC|8QH?<mG$6eBL5!eZD+HFFr%`1K(TS~vxDMwOx+
zJE7SjL#Geg4!TIuqmDA7AK?Eob&x*ZWPUaYP0zXuLW0nGTDxXr?kOQpvYu^Quvf>J
zCr_G%te1I#IW@H)sRx-?J;a=OQU~j)W!}7DQm*XQrlI6)4OrOZ!~8vSuyLH#VPOS(
z!IKcF&xlto5AlDSdvin3GXFZ{j^2Po_0WoESv{-A$~H{Pzt|9$b$D*Ua}zlH4JiEx
zR$Btr%OrMmi^rfr9#61pz)6_tkdMjRzN5oHy@n!Dr-1Pd@^Ot&;0JA4H|2_MT&$na
z+)2V3l5iX#ta-Ty+@931><;;q;rcYl6T6(=syKaar&DlU<r4s*fnaf3KT{yk<Ny|p
zV|f}&EfLzfE@c~8u2DV<8$Dp8VbjmCM)^E#YZjh-gGoap)CS`Vnj4sGUj6_R(ND0(
zT?xOso$y2@;fru8nqaL{zhoT0sTr{-74govsGmWUr!@&U47ot#ixyB}wpEaA`<Jt2
zO_~LQTb4s0d1FokFDIeG*cAhx2kR$+PCy__*2jEM*eDd{)-3Ybd1=G4@`3sS?a<uv
z^c{hAf_C~&PI$|lC$AA0T-8IG+qeiU%XcBv##5IYV)TorPGR(E(Os|<+`y);N~)7j
zce}zQKQQG*@G;5!B{AEqelsr=t!jidV?Xl1xag%BF_&$DT?}MfS&KntNf|>Ed58rJ
zGPc69P`0h2YXe=oX4PL}?Rb&_;fN)mgTNVS`DJ*<)KSfvCq-+)3VrOZBxlnPt262$
zoXHdQ;FbhFeRVk`gJ_(+4%PvuRg4P8KDi1~!3FgNzqY>4N>ZJ$c}|dFR|`%E+EcN(
zSRoposxEa@^~+JTV1Cv~d)G~H+PhOiE*nIpet>o21rq&P5sneoufg3`G=YqqFoD}p
z#qH^*Zi{A^$FMjb{)(_YF#oc#!VD|=Dl0&<<s;jx=Ee#lO!eaw>qa#+=n(5R?wt?|
zf!v!XtVcdyC%{b<W?>f8;I*(}g-bA|ev1WhH|YQ!Tod3X88Ye$f_B#6{I9>ZoIP8~
z<m4~cvz4!@0Pi$9{1q7+2^n0mDGg8`>@Dks7egP&a6jnd#`iVWXPo~D`5Nnjt7b$q
z%hM^6Fu@H2Y+-$HoXr-&Hwf>MF?gwu3kHQg+ORD*0F6EZ{(k|_XPQ@^E5RHUE(_tk
zn}4>yO2|2W-tkLvng5}-hX<X{kqBRMu9NHhk@Gf5^L5vciNZtfJh{z(?w$}ITKH^N
zfTz6<9`_t1CH}VO9{DYw@~)B(`Mcg*<WKx$%?0N(@Vnx^sMNuivX9@Z36W3u$2Bv%
z^!xc}Z6|+F<K!RLdih`Lt9Y)qt?Lk)pt*7w;V42Af)BwD@HwQKCWrZOT_^cFzg(9k
z{{)SXe_nT7{Exs#>eW`m{s<ZzMc8fs7*+=$-uQWp6*eiuxMmWe4*}zbMP&lzV+gy=
z;YJr$@b{Et7XN1_w#TBVw8Jn=H|HB{#2=}en_Y@6q!v?Z#F<!7W81^c2aU~Qhb@&>
z<H@+PqU-^yh?mZ(u_Yx#aoy}(N_~Zg{Yp*QHL<c1SJYG{eOys8iZaRH@~1py2UTWo
z#L{u~Mn;(r_ix&o1B4%K36fEsX!*_Lad5dK^L;oRLpXrYg0L5%6QLd<j4+OH5Me)$
z23m=Wvw)xfDDb=PCL+SeOeEvu5v51cf2C@>=pr^^ek{VrVcC46wSkBYCjQg5H-R$a
z?Jd2_=qn0PCaQ872MWRx!X<<_!Zgpd_Y(i+JMAu0(AWTMiLn9qiLxK%-MwGr<*w6`
z;V;9(-Tl5BxaJB%8lY@lxxSi6EX0*KU+m71CH|N0;d)F%`8GlXVIJWo?+kVYUqJB!
zK-s#QP9-wRqLM%rrV_R*jVSNqiQtmwbrg|-W%G&}UkYvhEoc|XE&h+*Nc$O7iz!;R
z%w<;7>Q<#0G%)BC<@2Go05+s7B~lA&Tv2Y}<T}D_gf;$7XqZg%PeQ}>ui-iqa6&Qm
z&Y<Rg9tii4H#Vojj|e%<kM-{(@AI4eRD3A%Px}M*lVI~9uu?h2Ed$?`vglLh-yf)L
zxeMY0<f{aNL>!MLVD>}&;jg^Se?M@X{Dez`W8^A7GT1(P9n~p-NNqB%#?l!YkIky8
zlDf94##fYAaK@flRbw=sRO0cimd^1X4)%}?e{b+)uw<-=(_}ohFncY&u(f%2(5-QJ
z8A$K)ks+=AJE&Vj_#VQW2#o)6Xdvgdxok;Wzk&&eYa-U!WIU}dy<%)#Sz4M+)7cf}
zC73K*(yNOwaTer=t#Tl4Dp%2!(K-<&gG2aFPQ-+}Yet;~;RgsWB4DJgOo3~>JXFos
H_5}YA!&h=j

diff --git a/__pycache__/orm.cpython-38.pyc b/__pycache__/orm.cpython-38.pyc
index e7f1513e7b5022c1453677dacaca495e5265a3af..56b7a560f897e388cd921f1712e6618bb3830b7d 100644
GIT binary patch
delta 721
zcmb7>&ubGw6vs20Y&N^eCfkarrA@3!sx|#(Q#~m71ECd7gP8WRNSD-^!or3wZ#RNo
zJhUgJAbEd+cqpEve})I|5xjWHLGceL;(O^KC=_vEJ~Q*)%)IY+=Eu~BqP=O`R)X5w
z?!Fv+v>zwMQ`}1a)F-65C$OEGzoV-$X~@i;CR0*B(A=W>J5o{?r6sd6C+&lzJ1z6F
zu%B=*NvD;V9mB(vs|}o<{#q>J8@(wWVqc#YWjxmJi#fcTKCVHA1oY25YlmLocl_83
zVRze$Uxcb(7zG3MTm`sh+$vwA?XdVKmUuZqxJo^`f$xoK(h=iFpNyms75toOlvgQp
zsKFH)T%#MQz>ng7`iY9VK@2k#h0A!`y!VQ4fEr_-F)B2y!Ns?!M;aj^UKU#IP8g^K
z&jS^L=QU?1VVN>eXRI*JGOjapMw&6lFbVzi&MtJ?v71t!FI~&WU#4HH4l{6~?Nq-N
z+SSCKRVRmEjVb(Qy<2JUNrS|&tqqu@B|KoP5muo{9Pp0=ABGLaBeGq`@7Y56-_~O+
z<w^_xV-VIc$ep}Coa%&C@8FkQRWy!-eNKpVypX?{9cttpEaO(b;*9=cI9tMAzGYUN
K^Uk;>Eb$vw8JTtf

delta 596
zcma)&JxhX76vw^l^Xl`c6{2Xq^=M(1W;zuFQ3=`$k=0<4N!KATkgH8?CNw#nv7sRd
zq$&IWeFS%xT3e!}t)Y8tkOsjE_s6;Sy#3F44t)CcBVAWb{CI0E`tYLfTSykpEMHS1
zwNjyNE_y3MWr9jnzBNO^y66i~g{ss>wYH@gq&l^?OvMm&6ihKEyjqJ85bj6<o`gM;
zgKr^B!r&G+NE}YYA6s9!DA!8W%d5)<G^JHziT7oR|48D~%Q4SyJp=DjlxSU9wh&@~
zrtHX@{J&#nb8xP>#3)A3!i}<S#L&bfwtyIx9aO|tWq4D9kraBD5ov@AkwFLu5fO(O
zwHc3avoEqr{kqy<Bc&QVY8QHHn#=RaehAqfGkMIEN`I+`Y#M835nCKt7U2BAr87)s
z2Z$V>Ar14IV~jcKaH<6p<II7T6~s1lv>gLiH+m9OT>+0CCHd}}?j~d#j_oV9fkyPP
Z6x`b*&f#JQRuX#lf?_y5&PkQ1<QD*?Z<_!B

diff --git a/controller_db.py b/controller_db.py
index c62fa65..fefb846 100644
--- a/controller_db.py
+++ b/controller_db.py
@@ -116,7 +116,8 @@ class ControllerDB:
                          n_max_trial, crit_supplier,
                          firm_req_prf_size, firm_req_prf_conn,
                          firm_acc_prf_size, firm_acc_prf_conn,
-                         netw_sply_prf_n, netw_sply_prf_size, cap_limit,
+                         netw_sply_prf_n, netw_sply_prf_size, 
+                         cap_limit_prob_type, cap_limit_level,
                          diff_new_conn, diff_remove, proactive_ratio):
         e = Experiment(
             idx_scenario=idx_scenario,
@@ -133,7 +134,8 @@ class ControllerDB:
             firm_acc_prf_conn=firm_acc_prf_conn,
             netw_sply_prf_n=netw_sply_prf_n,
             netw_sply_prf_size=netw_sply_prf_size,
-            cap_limit=cap_limit,
+            cap_limit_prob_type=cap_limit_prob_type,
+            cap_limit_level=cap_limit_level,
             diff_new_conn=diff_new_conn,
             diff_remove=diff_remove,
             proactive_ratio=proactive_ratio
diff --git a/model.py b/model.py
index 6b9aa4a..2b79815 100644
--- a/model.py
+++ b/model.py
@@ -21,7 +21,8 @@ class Model(ap.Model):
         self.int_n_max_trial = int(self.p.n_max_trial)
         self.flt_netw_sply_prf_n = float(self.p.netw_sply_prf_n)
         self.flt_netw_sply_prf_size = float(self.p.netw_sply_prf_size)
-        self.flt_cap_limit = int(self.p.cap_limit)
+        self.str_cap_limit_prob_type = str(self.p.cap_limit_prob_type)
+        self.flt_cap_limit_level = float(self.p.cap_limit_level)
         self.flt_diff_remove = float(self.p.diff_remove)
         self.proactive_ratio = float(self.p.proactive_ratio)
 
@@ -143,12 +144,25 @@ class Model(ap.Model):
                     code in attr['Product_Code']
                     for code in self.a_lst_total_products.code
                 ]))
-            # init extra capacity based on discrete uniform distribution
             for product in firm_agent.a_lst_product:
-                firm_agent.dct_prod_capacity[product] = \
-                    self.nprandom.integers(firm_agent.revenue_log / 5,
-                                           firm_agent.revenue_log / 5 +
-                                           self.flt_cap_limit)
+                # init extra capacity based on discrete uniform distribution
+                assert self.str_cap_limit_prob_type in ['uniform', 'normal'], \
+                    "cap_limit_prob_type other than uniform, normal"
+                if self.str_cap_limit_prob_type == 'uniform':
+                    extra_cap_mean = \
+                        firm_agent.revenue_log / self.flt_cap_limit_level
+                    extra_cap = self.nprandom.integers(extra_cap_mean-2,
+                                                       extra_cap_mean+2)
+                    extra_cap = 0 if round(extra_cap) < 0 else round(extra_cap)
+                    # print(firm_agent.name, extra_cap)
+                    firm_agent.dct_prod_capacity[product] = extra_cap
+                elif self.str_cap_limit_prob_type == 'normal':
+                    extra_cap_mean = \
+                        firm_agent.revenue_log / self.flt_cap_limit_level
+                    extra_cap = self.nprandom.normal(extra_cap_mean, 1)
+                    extra_cap = 0 if round(extra_cap) < 0 else round(extra_cap)
+                    # print(firm_agent.name, extra_cap)
+                    firm_agent.dct_prod_capacity[product] = extra_cap
 
             self.firm_network.add_agents([firm_agent], [ag_node])
         self.a_lst_total_firms = ap.AgentList(self, self.firm_network.agents)
@@ -277,7 +291,7 @@ class Model(ap.Model):
                             break
 
         # draw network
-        self.draw_network()
+        # self.draw_network()
 
     def update(self):
         self.a_lst_total_firms.clean_before_time_step()
diff --git a/orm.py b/orm.py
index b40cc32..0b561cf 100644
--- a/orm.py
+++ b/orm.py
@@ -62,7 +62,8 @@ class Experiment(Base):
     firm_acc_prf_conn = Column(Boolean, nullable=False)
     netw_sply_prf_n = Column(DECIMAL(8, 4), nullable=False)
     netw_sply_prf_size = Column(DECIMAL(8, 4), nullable=False)
-    cap_limit = Column(Integer, nullable=False)
+    cap_limit_prob_type = Column(String(16), nullable=False)
+    cap_limit_level = Column(DECIMAL(8, 4), nullable=False)
     diff_new_conn = Column(DECIMAL(8, 4), nullable=False)
     diff_remove = Column(DECIMAL(8, 4), nullable=False)
     proactive_ratio = Column(DECIMAL(8, 4), nullable=False)
diff --git a/xv_with_exp.csv b/xv_with_exp.csv
index 60acfd7..d6bd74a 100644
--- a/xv_with_exp.csv
+++ b/xv_with_exp.csv
@@ -1,4 +1,4 @@
-n_max_trial,crit_supplier,firm_req_prf_size,firm_req_prf_conn,firm_acc_prf_size,firm_acc_prf_conn,netw_sply_prf_size,netw_pref_supp_size,cap_limit,diff_new_conn,diff_remove,proactive_ratio
-15,2,2,True,2,True,0.5,2,4,0.5,0.5,0.4
-10,1,1,False,1,False,1,1,2,1,1,0.6
-5,0.5,0.5,,0.5,,2,0.5,1,2,2,0.8
+n_max_trial,crit_supplier,firm_req_prf_size,firm_req_prf_conn,firm_acc_prf_size,firm_acc_prf_conn,netw_sply_prf_size,netw_pref_supp_size,cap_limit_prob_type,cap_limit_level,diff_new_conn,diff_remove,proactive_ratio
+15,2,2,True,2,True,0.5,2,uniform,5,0.5,0.5,0.4
+10,1,1,False,1,False,1,1,normal,10,1,1,0.6
+5,0.5,0.5,,0.5,,2,0.5,,15,2,2,0.8
diff --git a/xv_without_exp.csv b/xv_without_exp.csv
index e9ca291..863c84b 100644
--- a/xv_without_exp.csv
+++ b/xv_without_exp.csv
@@ -1,2 +1,4 @@
-n_max_trial,crit_supplier,firm_req_prf_size,firm_req_prf_conn,firm_acc_prf_size,firm_acc_prf_conn,netw_sply_prf_n,netw_sply_prf_size,cap_limit,diff_new_conn,diff_remove,proactive_ratio
-10,1,1,True,1,True,1,1,2,1,1,0.2
+n_max_trial,crit_supplier,firm_req_prf_size,firm_req_prf_conn,firm_acc_prf_size,firm_acc_prf_conn,netw_sply_prf_n,netw_sply_prf_size,cap_limit_prob_type,cap_limit_level,diff_new_conn,diff_remove,proactive_ratio
+10,1,1,True,1,True,1,1,normal,10,1,1,0.2
+,,,,,,,,,,,,
+,,,,,,,,,,,,