Skip to content

Make additional model properties available through scivision interface #265

Open
@acocac

Description

@acocac

Feature Request

The current version of scivision only returns some features of the selected model. It'd be great to inform other relevant information such as number of parameters, model size, number of labels (for segmentation/classification models), etc. @darribas indicated about this feature as a feedback in a recent scivision demo.

Describe the solution you'd like

Adding more info when describing the model e.g. size

scivision.PretrainedModel( 
  module='', 
  model='', 
  source='' 
  pipe='' ,
  size= '',
  parameters=''
)

Describe alternatives you've considered

Number of labels (classification) or range of values (regression) could be a tricky information to extract from the model. We could ask to model contributors to add this information somewhere in the scivision_model.yml file?

Additional information

Metadata

Metadata

Assignees

No one assigned

    Labels

    enhancementNew feature or request

    Type

    No type

    Projects

    No projects

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions