snajmark commited on
Commit
4abe90f
·
1 Parent(s): 9b86f8d

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +7 -5
utils.py CHANGED
@@ -12,10 +12,11 @@ explainer_filename = "models/explainer_old.bz2"
12
 
13
  feature_names = ['PROPERTY: BCC/FCC/other', 'PROPERTY: Calculated Density (g/cm$^3$)',
14
  'PROPERTY: Calculated Young modulus (GPa)',
15
- 'PROPERTY: Processing method', 'PROPERTY: Microstructure',
16
- 'PROPERTY: Single/Multiphase', 'Microstructure One Hot',
17
- 'Processing Method One Hot', 'BCC/FCC/other One Hot',
18
- 'Single/Multiphase One Hot', 'Microstructure B2',
 
19
  'Microstructure B2+BCC', 'Microstructure B2+L12',
20
  'Microstructure B2+Laves+Sec.', 'Microstructure B2+Sec.',
21
  'Microstructure BCC', 'Microstructure BCC+B2',
@@ -42,7 +43,8 @@ feature_names = ['PROPERTY: BCC/FCC/other', 'PROPERTY: Calculated Density (g/cm$
42
  'Preprocessing method CAST', 'Preprocessing method OTHER',
43
  'Preprocessing method POWDER', 'Preprocessing method WROUGHT',
44
  'BCC/FCC/other BCC', 'BCC/FCC/other FCC', 'BCC/FCC/other OTHER',
45
- 'Single/Multiphase ', 'Single/Multiphase M', 'Single/Multiphase S']
 
46
 
47
  def return_feature_names():
48
  return feature_names
 
12
 
13
  feature_names = ['PROPERTY: BCC/FCC/other', 'PROPERTY: Calculated Density (g/cm$^3$)',
14
  'PROPERTY: Calculated Young modulus (GPa)',
15
+ #'PROPERTY: Processing method', 'PROPERTY: Microstructure',
16
+ #'PROPERTY: Single/Multiphase', 'Microstructure One Hot',
17
+ #'Processing Method One Hot', 'BCC/FCC/other One Hot',
18
+ #'Single/Multiphase One Hot',
19
+ 'Microstructure B2',
20
  'Microstructure B2+BCC', 'Microstructure B2+L12',
21
  'Microstructure B2+Laves+Sec.', 'Microstructure B2+Sec.',
22
  'Microstructure BCC', 'Microstructure BCC+B2',
 
43
  'Preprocessing method CAST', 'Preprocessing method OTHER',
44
  'Preprocessing method POWDER', 'Preprocessing method WROUGHT',
45
  'BCC/FCC/other BCC', 'BCC/FCC/other FCC', 'BCC/FCC/other OTHER',
46
+ #'Single/Multiphase ',
47
+ 'Single/Multiphase M', 'Single/Multiphase S']
48
 
49
  def return_feature_names():
50
  return feature_names