snajmark commited on
Commit
52b28f3
·
1 Parent(s): a22ee3e

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +3 -9
utils.py CHANGED
@@ -10,13 +10,8 @@ import matplotlib.pyplot as plt
10
  # Explainer path
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',
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,8 +38,7 @@ feature_names = ['PROPERTY: BCC/FCC/other', 'PROPERTY: Calculated Density (g/cm$
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
 
10
  # Explainer path
11
  explainer_filename = "models/explainer_old.bz2"
12
 
13
+ feature_names = ['PROPERTY: Calculated Density (g/cm$^3$)',
14
+ 'PROPERTY: Calculated Young modulus (GPa)', 'Microstructure B2',
 
 
 
 
 
15
  'Microstructure B2+BCC', 'Microstructure B2+L12',
16
  'Microstructure B2+Laves+Sec.', 'Microstructure B2+Sec.',
17
  'Microstructure BCC', 'Microstructure BCC+B2',
 
38
  'Preprocessing method CAST', 'Preprocessing method OTHER',
39
  'Preprocessing method POWDER', 'Preprocessing method WROUGHT',
40
  'BCC/FCC/other BCC', 'BCC/FCC/other FCC', 'BCC/FCC/other OTHER',
41
+ 'Single/Multiphase ', 'Single/Multiphase M', 'Single/Multiphase S']
 
42
 
43
  def return_feature_names():
44
  return feature_names