woshixuhao commited on
Commit
78d398a
·
1 Parent(s): 30c45be

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -315,7 +315,7 @@ def predict_xlsx(file):
315
  file_open.to_csv(file_name)
316
  return file_name
317
 
318
- def get_data_from_smile(smile):
319
  x_PE = np.array([[0, 1, 0, 0, 0], [0.333333, 0.666667, 0, 0, 0], [0.5, 0.5, 0, 0, 0],
320
  [0.75, 0.25, 0, 0, 0], [0.833333, 0.166667, 0, 0, 0], [0.952381, 0.047619, 0, 0, 0],
321
  [0.980392, 0.019608, 0, 0, 0], [1, 0, 0, 0, 0]], dtype=np.float32)
@@ -372,8 +372,8 @@ def convert_fig_PIL(fig):
372
  def predict_compare(smile_1,smile_2):
373
  config = parse_args()
374
  config.add_dipole = False
375
- X_test_PE_1,X_test_Et_1,X_test_ME_1=get_data_from_smile(smile_1)
376
- X_test_PE_2,X_test_Et_2,X_test_ME_2=get_data_from_smile(smile_2)
377
  Rf_all=[]
378
  for x_test in [X_test_PE_1,X_test_Et_1,X_test_ME_1,X_test_PE_2,X_test_Et_2,X_test_ME_2]:
379
  Model = Model_ML(config,x_test)
 
315
  file_open.to_csv(file_name)
316
  return file_name
317
 
318
+ def get_data_from_smile_compare(smile):
319
  x_PE = np.array([[0, 1, 0, 0, 0], [0.333333, 0.666667, 0, 0, 0], [0.5, 0.5, 0, 0, 0],
320
  [0.75, 0.25, 0, 0, 0], [0.833333, 0.166667, 0, 0, 0], [0.952381, 0.047619, 0, 0, 0],
321
  [0.980392, 0.019608, 0, 0, 0], [1, 0, 0, 0, 0]], dtype=np.float32)
 
372
  def predict_compare(smile_1,smile_2):
373
  config = parse_args()
374
  config.add_dipole = False
375
+ X_test_PE_1,X_test_Et_1,X_test_ME_1=get_data_from_smile_compare(smile_1)
376
+ X_test_PE_2,X_test_Et_2,X_test_ME_2=get_data_from_smile_compare(smile_2)
377
  Rf_all=[]
378
  for x_test in [X_test_PE_1,X_test_Et_1,X_test_ME_1,X_test_PE_2,X_test_Et_2,X_test_ME_2]:
379
  Model = Model_ML(config,x_test)