MaterialsProjectFeatureGenerator¶
-
class
mastml.feature_generators.
MaterialsProjectFeatureGenerator
(composition_df, api_key)[source]¶ Bases:
mastml.feature_generators.BaseGenerator
Class that wraps MaterialsProjectFeatureGeneration, giving it scikit-learn structure
- Args:
composition_df: (pd.DataFrame), dataframe containing vector of chemical compositions (strings) to generate elemental features from
mapi_key: (str), string denoting your Materials Project API key
- Methods:
- fit: pass through, copies input columns as pre-generated features
- Args:
- df: (dataframe), input dataframe containing X and y data
- transform: generate Materials Project features
- Args:
- df: (dataframe), input dataframe containing X and y data
- Returns:
- df: (dataframe), output dataframe containing generated features, original features and y data
Methods Summary
fit
(X[, y])generate_materialsproject_features
(X)transform
(X)Methods Documentation