Refactored activation function map to use custom type for lookup.