MaterialsProjectFeatureGenerator¶
- class mastml.feature_generators.MaterialsProjectFeatureGenerator(composition_df, api_key)[source]¶
Bases:
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])transform
(X)Methods Documentation