pulse2percept.datasets.greenwald2009

load_greenwald2009

Functions

load_greenwald2009([subjects, electrodes, ...])

Load data from [Greenwald2009]

pulse2percept.datasets.greenwald2009.load_greenwald2009(subjects=None, electrodes=None, shuffle=False, random_state=0)[source]

Load data from [Greenwald2009]

Load the brightness and size rating data described in [Greenwald2009]. Datapoints were extracted from figure 4 of the paper using WebplotDigitizer.

Retinal implants:

Argus I

Subjects:

2

Number of samples:

83

Number of features:

12

The dataset includes the following features:

subject

Subject ID, S06

implant

Argus I

electrode

Electrode ID, A2, A4, B1, C1, C4, D2, D3, D4

task

‘rate’

stim_class

“Greenwald2009”

stim_dur

Stimulus duration (ms)

amp

Amplitude of the stimulation

brightness

Patient reported brightness

pulse_dur

Pulse duration (ms)

interphase_dur

Interphase gap (ms)

pulse_type

‘cathodicFirst’

threshold

Electrical stimulation threshold

Added in version 0.7.

Parameters:
  • subjects (str | list of strings | None, optional) – Select data from a single subject or a list of subjects. By default, all subjects are selected

  • electrodes (str | list of strings | None, optional) – Select data from a single electrode or a list of electrodes. By default, all electrodes are selected.

  • shuffle (boolean, optional) – If True, the rows of the DataFrame are shuffled.

  • random_state (int | numpy.random.RandomState | None, optional) – Determines random number generation for dataset shuffling. Pass an int for reproducible output across multiple function calls.

Returns:

data – The whole dataset is returned in a 144x16 Pandas DataFrame

Return type:

pd.DataFrame