lucent.misc package

misc.io

• lucent.misc.io package
  • Submodules
  • lucent.misc.io.collapse_channels module
  • lucent.misc.io.serialize_array module
  • lucent.misc.io.showing module
  • Module contents

misc.channel_reducer module

Helper for using sklearn.decomposition on high-dimensional tensors.

Provides ChannelReducer, a wrapper around sklearn.decomposition to help them apply to arbitrary rank tensors. It saves lots of annoying reshaping.

class lucent.misc.channel_reducer.ChannelReducer(n_components: Optional[int] = 3, reduction_alg: Optional[Union[str, BaseEstimator]] = 'NMF', **kwargs)

Bases: object

fit(acts)

fit_transform(acts)

transform(acts)

Module contents