Skip to content

Commit

Permalink
show how to list models in readme
Browse files Browse the repository at this point in the history
  • Loading branch information
nreinicke committed Oct 26, 2023
1 parent 01c379d commit cd8eb35
Showing 1 changed file with 10 additions and 0 deletions.
10 changes: 10 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -49,6 +49,16 @@ Then, you can import the package and use a pre-trained model from the RouteE mod
import pandas as pd
import nrel.routee.powertrain as pt

# Print the available pre-trained models
print(pt.list_available_models(local=True, external=True))

# [
# '2016_TOYOTA_Camry_4cyl_2WD',
# '2017_CHEVROLET_Bolt',
# '2012_Ford_Focus',
# ...
# ]

# Load a pre-trained model
model = pt.load_model("2016_TOYOTA_Camry_4cyl_2WD")

Expand Down

0 comments on commit cd8eb35

Please sign in to comment.