Skip to content

Commit 7a469e4

Browse files
committed
Update figure: ESM in grey color
1 parent b8b324b commit 7a469e4

File tree

3 files changed

+7
-8
lines changed

3 files changed

+7
-8
lines changed
-4.98 KB
Loading

.gitignore

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -412,3 +412,4 @@ scripts/ESM_finetuning/results/dca_esm_and_hybrid_opt_results_clean.csv
412412
scripts/ESM_finetuning/results/dca_esm_and_hybrid_opt_results.csv
413413
scripts/ESM_finetuning/mut_performance.png
414414
scripts/ESM_finetuning/_Description_DMS_substitutions_data.csv
415+
scripts/ESM_finetuning/mut_performance_violin.png

scripts/ESM_finetuning/run_performance_tests_proteingym_data.py

Lines changed: 6 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -351,7 +351,7 @@ def plot_csv_data(csv, plot_name):
351351
'Hybrid_Trained_Performance_1000'
352352
]]
353353
print(df_)
354-
sns.violinplot(df_, saturation=0.4)
354+
sns.violinplot(df_, saturation=0.4, palette=['tab:blue', 'tab:grey', 'tab:orange', 'tab:green', 'tab:red'])
355355
plt.ylim(-0.09, 1.09)
356356
plt.ylabel(r'|Spearmanr $\rho$|')
357357
sns.swarmplot(df_, color='black')
@@ -431,18 +431,16 @@ def plot_csv_data(csv, plot_name):
431431
already_tested_is = []
432432

433433

434-
compute_performances(
435-
mut_data=combined_mut_data,
436-
start_i=start_i,
437-
already_tested_is=already_tested_is
438-
)
434+
#compute_performances(
435+
# mut_data=combined_mut_data,
436+
# start_i=start_i,
437+
# already_tested_is=already_tested_is
438+
#)
439439

440440

441441
with open(out_results_csv, 'r') as fh:
442442
with open(os.path.join(os.path.dirname(__file__), 'results/dca_esm_and_hybrid_opt_results_clean.csv'), 'w') as fh2:
443443
for line in fh:
444-
#print(line.split(','))
445-
print(line.split(',')[1])
446444
if not line.split(',')[1].startswith('OOM') and not line.split(',')[1].startswith('X'):
447445
fh2.write(line)
448446

0 commit comments

Comments
 (0)