Skip to content

Add ART1 algorithm in neural network #12321

Open
@JeninaAngelin

Description

Feature description

Description:

This issue aims to implement an Adaptive Resonance Theory (ART1) algorithm for binary data clustering. ART1 is well-suited for unsupervised learning with binary inputs, using a vigilance parameter to determine the clustering threshold. This model will enhance the framework's clustering capabilities, allowing for pattern recognition in binary data.

Goals:

  • Implement an ART1 class with training and prediction functionalities.
  • Allow the user to specify the vigilance parameter to control cluster formation.
  • Include a similarity calculation and weight adjustment function.

    Requirements:

  • Initialization: ART1 class should initialize with num_features and vigilance.
  • Training: Implement a train(data) method to cluster binary data based on the vigilance parameter.
  • Prediction: Add a predict(x) method to classify new input into an existing cluster or mark it as a new cluster if it does not match any.
  • Documentation: Include docstrings and usage examples to clarify the purpose of each method.
  • Testing: Provide example usage for verification and add unit tests to confirm functionality.
  • Activity

    Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

    Metadata

    Assignees

    No one assigned

      Labels

      enhancementThis PR modified some existing files

      Type

      No type

      Projects

      No projects

      Milestone

      No milestone

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions