feature_importance.json

{
  "feature_names": [
    "avg_line_length",
    "compute_bound_score",
    "has_attention",
    "has_backward",
    "has_batchnorm",
    "has_cat",
    "has_conv",
    "has_dropout",
    "has_embedding",
    "has_fft",
    "has_gelu",
    "has_layernorm",
    "has_linear",
    "has_loss",
    "has_matmul",
    "has_pooling",
    "has_reduce",
    "has_relu",
    "has_reshape",
    "has_sigmoid",
    "has_softmax",
    "has_sort",
    "has_tanh",
    "has_transpose",
    "max_dim",
    "max_numeric",
    "mean_numeric",
    "memory_bound_score",
    "min_numeric",
    "mixed_score",
    "num_chars",
    "num_class_defs",
    "num_dim_specs",
    "num_for_loops",
    "num_function_defs",
    "num_imports",
    "num_large_dims",
    "num_lines",
    "num_nn_calls",
    "num_numeric_literals",
    "num_tokens",
    "num_torch_calls",
    "product_large_dims_log",
    "sum_numeric_log",
    "total_elements_log",
    "uses_cuda",
    "uses_float16",
    "uses_float32",
    "cuda_cores",
    "tensor_cores",
    "memory_gb",
    "memory_bandwidth_gbps",
    "base_clock_mhz",
    "boost_clock_mhz",
    "sm_count",
    "fp32_tflops",
    "fp16_tflops",
    "tdp_watts",
    "compute_capability",
    "l2_cache_mb",
    "flops_log",
    "memory_bytes_log",
    "arithmetic_intensity",
    "dtype_bytes"
  ],
  "gbr_importance": [
    0.00020907692527896766,
    1.8562661189242273e-07,
    2.9369056959320215e-05,
    0.0,
    0.0,
    0.0,
    1.34552169063225e-06,
    0.0,
    1.2959966275380288e-08,
    2.8763114459042767e-08,
    9.104787556291898e-06,
    5.54221245471016e-06,
    4.32329418665271e-06,
    0.0,
    1.2805806474662726e-06,
    2.257546676504885e-08,
    1.735686571541027e-07,
    8.901842573818885e-07,
    2.311281409817864e-06,
    9.20361423082855e-07,
    2.7881975535483807e-06,
    2.0856182994836325e-08,
    1.1771724567239878e-06,
    2.7156362706913446e-05,
    8.131880276961555e-05,
    7.180057550667766e-05,
    0.0004996830995928254,
    1.5779269357750438e-06,
    0.0001825415270437779,
    1.6970423039960086e-05,
    0.00041935952304566224,
    3.4117339243641743e-06,
    6.432292741146238e-06,
    0.0,
    5.033034070243649e-06,
    1.945656188484425e-06,
    4.355968723707434e-05,
    2.3658129881027048e-05,
    7.174538256656928e-06,
    5.067777589608545e-05,
    4.413329058621823e-05,
    6.258530983657731e-06,
    0.0003196131073257341,
    0.0009352059938410164,
    0.005090926310276446,
    0.0,
    0.01927022907891746,
    0.02657711487616414,
    0.01086756094557264,
    0.003634432430735974,
    0.0007652333385088071,
    0.007919310415035623,
    0.001091000073631452,
    0.012641277527748929,
    0.010506180230114641,
    0.010371283017597741,
    0.015597399741008567,
    0.002644217662689972,
    0.008205515687632867,
    0.0018993846795010236,
    0.6869183995385755,
    0.1375995391029925,
    0.009656091846671428,
    0.0257288175591774
  ]
}