dcgan.py: `df.plot(x=np.array([1, 101, 201]))` doesn't work

Line https://github.com/pytorch/ignite/blob/master/examples/gan/dcgan.py#L393 fails with an error,saying that nothing in Int64Index([1, 101, 201]) is in df.columns (or something like that). To fix it, set index of the dataframe instead of passing x argument.

Replace this:

df = pd.read_csv(os.path.join(output_dir, LOGS_FNAME), delimiter='\t')
x = np.arange(1, engine.state.iteration + 1, PRINT_FREQ)
_ = df.plot(x=x, subplots=True, figsize=(20, 20))

with this

df = pd.read_csv(os.path.join(output_dir, LOGS_FNAME), delimiter='\t') \
    .set_index(np.arange(1, engine.state.iteration + 1, PRINT_FREQ))
_ = df.plot(subplots=True, figsize=(20, 20))

My python version is 3.7, pandas is 0.24.2.

[philip-bl]

@vfdev-5 Not really. Overhead of doing git clone, setting up remote repositories, applying patch there, pushing, creating a PR - it’s too much.

[philip-bl]

@vfdev-5 Yes, that sounds better. Just to be clear, I don’t want to do be the person responsible for implementing that change.