Skip to content

Commit 60f8b86

Browse files
committed
Improved bestK text.
1 parent ca9d6f4 commit 60f8b86

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

structure_threader/wrappers/maverick_wrapper.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -132,17 +132,21 @@ def ti_test(outdir, norm_evidence, ti_in_use):
132132
if ti_in_use:
133133
# Use TI for bestK estimation
134134
criteria = norm_evidence[2]
135+
estimator = "TI"
135136
else:
136137
# Use Structure for bestK estimation
137138
criteria = norm_evidence[1]
139+
estimator = "STRUCTURE"
138140
means = {x: y["norm_mean"] for x, y in criteria.items()}
139141

140142
bestk_dir = os.path.join(outdir, "bestK")
141143
os.makedirs(bestk_dir, exist_ok=True)
142144
bestk = max(means, key=means.get)
143145
bestk_file = open(os.path.join(bestk_dir, "TI_integration.txt"), "w")
144-
output_text = ("MavericK's estimation test revealed "
145-
"that the best value of 'K' is: {}\n".format(bestk))
146+
output_text = ("MavericK's estimation test ({}) revealed that the best "
147+
"value of 'K' is: {}\nIt is still recommended that you look "
148+
"at the generated plot for more accurate "
149+
"information".format(estimator, bestk))
146150
bestk_file.write(output_text)
147151
bestk_file.close()
148152

0 commit comments

Comments
 (0)