transpolymer commited on
Commit
dd5919f
·
verified ·
1 Parent(s): 6bb0796

Update prediction.py

Browse files
Files changed (1) hide show
  1. prediction.py +1 -1
prediction.py CHANGED
@@ -12,7 +12,7 @@ embedding_model = AutoModel.from_pretrained("seyonec/ChemBERTa-zinc-base-v1")
12
 
13
  # Load saved scalers (for inverse_transform)
14
  scaler_tensile_strength = joblib.load("scaler_Tensile_strength_Mpa_.joblib") # Scaler for Tensile Strength
15
- scaler_ionization_energy = joblib.load("scaler_lonization_Energy_eV_.joblib") # Scaler for Ionization Energy
16
  scaler_electron_affinity = joblib.load("scaler_Electron_Affinity_eV_.joblib") # Scaler for Electron Affinity
17
  scaler_logp = joblib.load("scaler_LogP.joblib") # Scaler for LogP
18
  scaler_refractive_index = joblib.load("scaler_Refractive_Index.joblib") # Scaler for Refractive Index
 
12
 
13
  # Load saved scalers (for inverse_transform)
14
  scaler_tensile_strength = joblib.load("scaler_Tensile_strength_Mpa_.joblib") # Scaler for Tensile Strength
15
+ scaler_ionization_energy = joblib.load("scaler_Ionization_Energy_eV_.joblib") # Scaler for Ionization Energy
16
  scaler_electron_affinity = joblib.load("scaler_Electron_Affinity_eV_.joblib") # Scaler for Electron Affinity
17
  scaler_logp = joblib.load("scaler_LogP.joblib") # Scaler for LogP
18
  scaler_refractive_index = joblib.load("scaler_Refractive_Index.joblib") # Scaler for Refractive Index