ElementalFeatureGenerator¶
- class mastml.feature_generators.ElementalFeatureGenerator(composition_df, feature_types=None, remove_constant_columns=False)[source]¶
Bases:
BaseGenerator
Class that is used to create elemental-based features from material composition strings
- Args:
composition_df: (pd.DataFrame), dataframe containing vector of chemical compositions (strings) to generate elemental features from
feature_types: (list), list of strings denoting which elemental feature types to include in the final feature matrix.
remove_constant_columns: (bool), whether to remove constant columns from the generated feature set
- Methods:
- fit: pass through, copies input columns as pre-generated features
- Args:
X: (pd.DataFrame), input dataframe containing X data
y: (pd.Series), series containing y data
- transform: generate the elemental feature matrix from composition strings
- Args:
None.
- Returns:
X: (dataframe), output dataframe containing generated features
y: (series), output y data as series
Methods Summary
fit
([X, y])transform
([X])Methods Documentation