OneHotElementEncoder

class mastml.feature_generators.OneHotElementEncoder(composition_df, remove_constant_columns=False)

Bases: BaseGenerator

Class to generate new categorical features (i.e. values of 1 or 0) based on whether an input composition contains a certain designated element

Args:

composition_df: (pd.DataFrame or pd.Series), dataframe containing vector of chemical compositions (strings) to generate elemental features from

remove_constant_columns: (bool), whether to remove constant columns from the generated feature set

Methods:

fit: pass through, needed to maintain scikit-learn class structure

Args:

df: (dataframe), dataframe of input X and y data

transform: generate new element-specific features

Args:

df: (dataframe), dataframe of input X and y data

Returns:

df_trans: (dataframe), dataframe with generated element-specific features

Methods Summary

fit(X[, y])
transform(X[, y])

Methods Documentation

fit(X, y=None)

transform(X, y=None)