Flexible modeling of neuron firing rates across different experimental conditions: an application to neural activity in the prefrontal cortex during a discriminatino task