Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[FEA]: Remove hard-coded fields from GenerateVizFramesStage #2136

Open
2 tasks done
dagardner-nv opened this issue Jan 27, 2025 · 0 comments
Open
2 tasks done

[FEA]: Remove hard-coded fields from GenerateVizFramesStage #2136

dagardner-nv opened this issue Jan 27, 2025 · 0 comments
Labels
feature request New feature or request

Comments

@dagardner-nv
Copy link
Contributor

Is this a new feature, an improvement, or a change to existing functionality?

Improvement

How would you describe the priority of this feature request

Medium

Please provide a clear description of problem this feature solves

The GenerateVizFramesStage (python/morpheus/morpheus/stages/postprocess/generate_viz_frames_stage.py) contains hard-coded fields specific to the SID pipeline ex:

idx2label = {
    0: 'address',
    1: 'bank_acct',
    2: 'credit_card',
    3: 'email',
    4: 'govt_id',
    5: 'name',
    6: 'password',
    7: 'phone_num',
    8: 'secret_keys',
    9: 'user'
}

columns = ["timestamp", "src_ip", "dest_ip", "src_port", "dest_port", "data"]

Describe your ideal solution

The idx2label values should be fetched from the Config.class_labels attribute, and the DF field names could be constructor arguments.

Alternately this class could be moved to examples/sid_visualization.

Additional context

No response

Code of Conduct

  • I agree to follow this project's Code of Conduct
  • I have searched the open feature requests and have found no duplicates for this feature request
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
feature request New feature or request
Projects
Status: Todo
Development

No branches or pull requests

1 participant