dodox/CaliGraph
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

No more stability-metric for the nn (but se instead)

Browse Source
This commit is contained in:
Dominik Moritz Roth 2021-09-24 17:35:32 +02:00
parent ec8d253f3a
commit f48fb12f0a
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
caliGraph.py
View File

@ -354,8 +354,10 @@ def scoreUnread(G, globMu, globStd, errorFac=-0.6):
node['se'] = globStd / math.sqrt(len(feedbacks)) node['se'] = globStd / math.sqrt(len(feedbacks))
feedbacks.append(node['std']) feedbacks.append(node['std'])
weights.append(getWeightForType('sigma')) weights.append(getWeightForType('sigma'))
feedbacks.append(1-1/len(feedbacks)) feedbacks.append(node['se'])
weights.append(getWeightForType('stability')) weights.append(getWeightForType('se'))
#feedbacks.append(1-1/len(feedbacks))
#weights.append(getWeightForType('stability'))
node['mean'] = sum([fb*w for fb, w in zip(feedbacks, weights)])/len(feedbacks) node['mean'] = sum([fb*w for fb, w in zip(feedbacks, weights)])/len(feedbacks)
node['score'] = node['mean'] + errorFac*node['se'] node['score'] = node['mean'] + errorFac*node['se']
else: else:
@ -716,7 +718,7 @@ def train(gamma = 0.1):
weights[attr] = -0.1+random.random()*1.5 weights[attr] = -0.1+random.random()*1.5
else: else:
weights[attr] += delta weights[attr] += delta
if attr not in ['sigma', 'mu', 'stability']: if attr not in ['sigma', 'mu', 'se']:
weights[attr] = min(max(0, weights[attr]), 3) weights[attr] = min(max(0, weights[attr]), 3)
mse = evaluateFitness(books) mse = evaluateFitness(books)
if mse < best_mse: # got better if mse < best_mse: # got better