neurox

Subpackages:

• neurox.analysis
  • neurox.analysis.corpus
    • get_top_words()
  • neurox.analysis.plotting
    • plot_accuracies_per_tag()
    • plot_distributedness()
    • plot_accuracies()
  • neurox.analysis.visualization
    • visualize_activations()
    • TransformersVisualizer
      • TransformersVisualizer.model_name
      • TransformersVisualizer.model
      • TransformersVisualizer.tokenizer
      • TransformersVisualizer.__call__()
      • TransformersVisualizer.__init__()
• neurox.data
  • neurox.data.extraction
    • neurox.data.extraction.transformers_extractor
      • get_model_and_tokenizer()
      • aggregate_repr()
      • extract_sentence_representations()
      • extract_representations()
      • main()
  • neurox.data.annotate
    • annotate_data()
  • neurox.data.control_task
    • create_sequence_labeling_dataset()
  • neurox.data.loader
    • load_activations()
    • filter_activations_by_layers()
    • load_aux_data()
    • load_data()
    • load_sentence_data()
  • neurox.data.representations
    • bpe_get_avg_activations()
    • bpe_get_last_activations()
    • char_get_avg_activations()
    • char_get_last_activations()
    • sent_get_last_activations()
  • neurox.data.utils
    • save_files()
  • neurox.data.writer
    • ActivationsWriter
      • ActivationsWriter.filename
      • ActivationsWriter.filetype
      • ActivationsWriter.decompose_layers
      • ActivationsWriter.filter_layers
      • ActivationsWriter.__init__()
      • ActivationsWriter.open()
      • ActivationsWriter.write_activations()
      • ActivationsWriter.close()
      • ActivationsWriter.get_writer()
      • ActivationsWriter.add_writer_options()
    • ActivationsWriterManager
      • ActivationsWriterManager.__init__()
      • ActivationsWriterManager.open()
      • ActivationsWriterManager.write_activations()
      • ActivationsWriterManager.close()
    • HDF5ActivationsWriter
      • HDF5ActivationsWriter.__init__()
      • HDF5ActivationsWriter.open()
      • HDF5ActivationsWriter.write_activations()
      • HDF5ActivationsWriter.close()
    • JSONActivationsWriter
      • JSONActivationsWriter.__init__()
      • JSONActivationsWriter.open()
      • JSONActivationsWriter.write_activations()
      • JSONActivationsWriter.close()
• neurox.interpretation
  • neurox.interpretation.ablation
    • keep_specific_neurons()
    • filter_activations_keep_neurons()
    • filter_activations_remove_neurons()
    • zero_out_activations_keep_neurons()
    • zero_out_activations_remove_neurons()
    • filter_activations_by_layers()
  • neurox.interpretation.clustering
    • create_correlation_clusters()
    • extract_independent_neurons()
    • print_clusters()
    • scikit_extract_independent_neurons()
  • neurox.interpretation.gaussian_probe
    • GaussianProbe
      • GaussianProbe.__init__()
    • train_probe()
    • evaluate_probe()
    • get_neuron_ordering()
  • neurox.interpretation.iou_probe
    • get_neuron_ordering()
  • neurox.interpretation.linear_probe
    • LinearProbe
      • LinearProbe.__init__()
      • LinearProbe.forward()
      • LinearProbe.training
    • l1_penalty()
    • l2_penalty()
    • train_logistic_regression_probe()
    • train_linear_regression_probe()
    • evaluate_probe()
    • get_top_neurons()
    • get_top_neurons_hard_threshold()
    • get_bottom_neurons()
    • get_random_neurons()
    • get_neuron_ordering()
    • get_neuron_ordering_granular()
    • get_fixed_number_of_bottom_neurons()
  • neurox.interpretation.metrics
    • accuracy()
    • f1()
    • accuracy_and_f1()
    • pearson()
    • spearman()
    • pearson_and_spearman()
    • matthews_corrcoef()
    • compute_score()
  • neurox.interpretation.probeless
    • get_neuron_ordering()
    • get_neuron_ordering_for_tag()
    • get_neuron_ordering_for_all_tags()
  • neurox.interpretation.utils
    • isnotebook()
    • get_progress_bar()
    • batch_generator()
    • tok2idx()
    • idx2tok()
    • count_target_words()
    • create_tensors()
    • print_overall_stats()
    • print_machine_stats()
    • balance_binary_class_data()
    • balance_multi_class_data()
    • load_probe()
    • save_probe()

Module contents: