Skip to content

Commit 7c6b925

Browse files
authored
fix: Change generator predict call to implicit call
1 parent a2c9a29 commit 7c6b925

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

examples/regular/gan_example.ipynb

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1322,13 +1322,13 @@
13221322
" ax = plt.subplot(rows, columns, model_step_ix*columns + 1 + (i+1) )\n",
13231323
"\n",
13241324
" if with_class:\n",
1325-
" g_z = generator_model.predict([z, labels])\n",
1325+
" g_z = generator_model([z, labels])\n",
13261326
" gen_samples = DataFrame(g_z, columns=num_cols+class_labels)\n",
13271327
" for group, color, marker, label in zip( gen_samples.groupby('Class_1'), colors, markers, class_labels ):\n",
13281328
" plt.scatter( group[1][[col1]], group[1][[col2]],\n",
13291329
" label=label, marker=marker, edgecolors=color, facecolors='none' )\n",
13301330
" else:\n",
1331-
" g_z = generator_model.predict(z)\n",
1331+
" g_z = generator_model(z)\n",
13321332
" gen_samples = DataFrame(g_z, columns=num_cols+class_labels)\n",
13331333
" gen_samples.to_csv('../../data/Generated_sample.csv')\n",
13341334
" plt.scatter( gen_samples[[col1]], gen_samples[[col2]],\n",

0 commit comments

Comments
 (0)