cz-ye commited on
Commit
bd3d7ad
1 Parent(s): e9744b4
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -4,8 +4,8 @@ from os import listdir
4
  import plotly.graph_objects as go
5
 
6
  gene_names_eve = [file.split('_')[0] for file in listdir('./CPT1_score_EVE_set/')]
7
- gene_names_no_eve1 = [file.split('_')[0] for file in listdir('./CPT1_score_no_EVE_set1/')]
8
- gene_names_no_eve2 = [file.split('_')[0] for file in listdir('./CPT1_score_no_EVE_set2/')]
9
 
10
  st.subheader('CPT-1')
11
  st.markdown('Cross-protein transfer learning for variant effect prediction')
@@ -21,9 +21,9 @@ if st.button('Show results'):
21
  if gene in gene_names_eve:
22
  pred = pd.read_csv('./CPT1_score_EVE_set/' + gene + '_HUMAN.csv.gz', compression = 'gzip')
23
  elif gene in gene_names_no_eve1:
24
- pred = pd.read_csv('./CPT1_score_no_EVE_set1/' + gene + '_HUMAN.csv.gz', compression = 'gzip')
25
  else:
26
- pred = pd.read_csv('./CPT1_score_no_EVE_set2/' + gene + '_HUMAN.csv.gz', compression = 'gzip')
27
 
28
  # Plot heatmap
29
  mat = pred.copy()
 
4
  import plotly.graph_objects as go
5
 
6
  gene_names_eve = [file.split('_')[0] for file in listdir('./CPT1_score_EVE_set/')]
7
+ gene_names_no_eve1 = [file.split('_')[0] for file in listdir('./CPT1_score_no_EVE_set_1/')]
8
+ gene_names_no_eve2 = [file.split('_')[0] for file in listdir('./CPT1_score_no_EVE_set_2/')]
9
 
10
  st.subheader('CPT-1')
11
  st.markdown('Cross-protein transfer learning for variant effect prediction')
 
21
  if gene in gene_names_eve:
22
  pred = pd.read_csv('./CPT1_score_EVE_set/' + gene + '_HUMAN.csv.gz', compression = 'gzip')
23
  elif gene in gene_names_no_eve1:
24
+ pred = pd.read_csv('./CPT1_score_no_EVE_set_1/' + gene + '_HUMAN.csv.gz', compression = 'gzip')
25
  else:
26
+ pred = pd.read_csv('./CPT1_score_no_EVE_set_2/' + gene + '_HUMAN.csv.gz', compression = 'gzip')
27
 
28
  # Plot heatmap
29
  mat = pred.copy()