-
Notifications
You must be signed in to change notification settings - Fork 35
Description
There is a problem in the visualization part. When I am trying to plot the graph in the example, I see the following mistake:
TypeError Traceback (most recent call last)
in
12 ice_plot(ice_df, frac_to_plot=0.1,
13 color_by='x3', cmap=PuOr,
---> 14 ax=ice_ax);
15
16 ice_ax.set_xlabel('$X_2$');
C:\ProgramData\Anaconda3\lib\site-packages\pycebox\ice.py in ice_plot(ice_data, frac_to_plot, plot_points, point_kwargs, x_quantile, plot_pdp, centered, centered_quantile, color_by, cmap, ax, pdp_kwargs, **kwargs)
128 if frac_to_plot < 1.:
129 n_cols = ice_data.shape[1]
--> 130 icols = np.random.choice(n_cols, size=frac_to_plot * n_cols, replace=False)
131 plot_ice_data = ice_data.iloc[:, icols]
132 else:
mtrand.pyx in mtrand.RandomState.choice()
TypeError: 'float' object cannot be interpreted as an integer