mgyigit commited on
Commit
82d01e1
·
verified ·
1 Parent(s): 35b9102

Update src/bin/binding_affinity_estimator.py

Browse files
src/bin/binding_affinity_estimator.py CHANGED
@@ -3,6 +3,12 @@ import multiprocessing
3
  import pandas as pd
4
  import numpy as np
5
  import scipy.stats
 
 
 
 
 
 
6
 
7
  from sklearn import linear_model
8
  from sklearn.model_selection import KFold
@@ -154,7 +160,8 @@ def predictAffinityWithModel(regressor_model,multiplied_vectors_df):
154
  validation_corr_pval_list,
155
  )
156
 
157
- ppi_affinity_file = "../data/auxilary_input/skempi_pipr/SKEMPI_all_dg_avg.txt"
 
158
  ppi_affinity_df = pd.read_csv(ppi_affinity_file,sep="\t",header=None)
159
  ppi_affinity_df.columns = ['Protein1', 'Protein2', 'Affinity']
160
 
 
3
  import pandas as pd
4
  import numpy as np
5
  import scipy.stats
6
+ import os
7
+ import sys
8
+
9
+ script_dir = os.path.dirname(os.path.abspath(__file__))
10
+ sys.path.append('..')
11
+ sys.path.append('.')
12
 
13
  from sklearn import linear_model
14
  from sklearn.model_selection import KFold
 
160
  validation_corr_pval_list,
161
  )
162
 
163
+ ppi_affinity_file_path = "../data/auxilary_input/skempi_pipr/SKEMPI_all_dg_avg.txt"
164
+ ppi_affinity_file = os.path.join(script_dir, ppi_affinity_file_ath)
165
  ppi_affinity_df = pd.read_csv(ppi_affinity_file,sep="\t",header=None)
166
  ppi_affinity_df.columns = ['Protein1', 'Protein2', 'Affinity']
167