car-data / constants.py
unixpickle's picture
model
ddd296c
PRICE_CUTOFFS = [
10_000.0,
15_000.0,
20_000.0,
25_000.0,
30_000.0,
35_000.0,
40_000.0,
50_000.0,
60_000.0,
]
NUM_PRICE_BINS = len(PRICE_CUTOFFS) + 1
PRICE_BIN_LABELS = [
"$0-$10,000",
"$10,000-$15,000",
"$15,000-$20,000",
"$20,000-$25,000",
"$25,000-$30,000",
"$30,000-$35,000",
"$35,000-$40,000",
"$40,000-$50,000",
"$50,000-$60,000",
"$60,000+",
]
MEDIAN_PRICE_SCALE = 30000.0
YEARS = list(range(1983, 2024))
NUM_YEARS = len(YEARS) + 1
MAKES_MODELS = (
("Ford", "F150"),
("Chevrolet", "Silverado 1500"),
("RAM", "1500"),
("Jeep", "Wrangler"),
("Ford", "Explorer"),
("Nissan", "Rogue"),
("Jeep", "Grand Cherokee"),
("Chevrolet", "Equinox"),
("GMC", "Sierra 1500"),
("Ford", "Escape"),
("Honda", "Accord"),
("Toyota", "Camry"),
("Toyota", "RAV4"),
("Honda", "Civic"),
("Honda", "CR-V"),
("MAZDA", "CX-5"),
("Toyota", "Tacoma"),
("Ford", "F250"),
("Toyota", "Corolla"),
("Toyota", "Highlander"),
("Jeep", "Cherokee"),
("Nissan", "Altima"),
("Subaru", "Outback"),
("RAM", "2500"),
("Honda", "Pilot"),
("Chevrolet", "Malibu"),
("Hyundai", "Tucson"),
("Ford", "Mustang"),
("Chevrolet", "Traverse"),
("Hyundai", "Santa Fe"),
("Hyundai", "Elantra"),
("Jeep", "Compass"),
("Chevrolet", "Silverado 2500"),
("Ford", "Edge"),
("Nissan", "Frontier"),
("Chevrolet", "Tahoe"),
("GMC", "Terrain"),
("Toyota", "Tundra"),
("GMC", "Acadia"),
("Volkswagen", "Tiguan"),
("Hyundai", "Sonata"),
("Subaru", "Forester"),
("Jeep", "Gladiator"),
("Chevrolet", "Colorado"),
("Nissan", "Pathfinder"),
("Toyota", "4Runner"),
("Ford", "Fusion"),
("Nissan", "Sentra"),
("Kia", "Sorento"),
("GMC", "Sierra 2500"),
("Ford", "F350"),
("Subaru", "Crosstrek"),
("Kia", "Sportage"),
("Honda", "HR-V"),
("Kia", "Forte"),
("Honda", "Odyssey"),
("Ford", "Bronco Sport"),
("Dodge", "Challenger"),
("Dodge", "Charger"),
("Buick", "Enclave"),
("Chevrolet", "Blazer"),
("Acura", "MDX"),
("Audi", "Q5"),
("Volkswagen", "Atlas"),
("Buick", "Envision"),
("Kia", "Soul"),
("Chrysler", "Pacifica"),
("Hyundai", "Kona"),
("Chevrolet", "Camaro"),
("Jeep", "Grand Cherokee L"),
("MAZDA", "CX-9"),
("Dodge", "Durango"),
("Nissan", "Murano"),
("Chevrolet", "Trax"),
("GMC", "Yukon"),
("Volkswagen", "Jetta"),
("BMW", "X5"),
("Chevrolet", "Suburban"),
("Ford", "Expedition"),
("Nissan", "Rogue Sport"),
("RAM", "3500"),
("Ford", "Bronco"),
("Honda", "Ridgeline"),
("Chevrolet", "Corvette"),
("Cadillac", "XT5"),
("Toyota", "Sienna"),
("Mitsubishi", "Outlander"),
("Kia", "Telluride"),
("Buick", "Encore"),
("Mercedes-Benz", "C 300"),
("BMW", "X3"),
("Subaru", "Ascent"),
("Honda", "Passport"),
("MAZDA", "MAZDA3"),
("Buick", "Encore GX"),
("Volvo", "XC90"),
("Mercedes-Benz", "GLC 300"),
("Ford", "Ranger"),
("Jeep", "Renegade"),
("Lexus", "RX 350"),
("Volvo", "XC60"),
("Kia", "Optima"),
("Chevrolet", "Silverado 3500"),
("Dodge", "Grand Caravan"),
("INFINITI", "QX60"),
("Nissan", "Titan"),
("Subaru", "WRX"),
("GMC", "Canyon"),
("Tesla", "Model 3"),
("Chevrolet", "Cruze"),
("Lexus", "ES 350"),
("Nissan", "Armada"),
("GMC", "Yukon XL"),
("GMC", "Sierra 3500"),
("Hyundai", "Palisade"),
("Ford", "Focus"),
("Kia", "Niro"),
("Toyota", "Prius"),
("INFINITI", "QX80"),
("Porsche", "Macan"),
("Chevrolet", "TrailBlazer"),
("Cadillac", "XT4"),
("MAZDA", "CX-50"),
("Lincoln", "Corsair"),
("Audi", "Q7"),
("Ford", "Expedition Max"),
("Cadillac", "Escalade"),
("MINI", "Cooper"),
("Acura", "RDX"),
("Subaru", "Impreza"),
("Audi", "A4"),
("Nissan", "Kicks"),
("Nissan", "Maxima"),
("Porsche", "Cayenne"),
("Dodge", "Journey"),
("Porsche", "911"),
("RAM", "ProMaster"),
("Mercedes-Benz", "GLE 350"),
("Ford", "EcoSport"),
("Volkswagen", "Taos"),
("MAZDA", "CX-30"),
("Lincoln", "Nautilus"),
("Land Rover", "Range Rover"),
("Mitsubishi", "Outlander Sport"),
("Lexus", "GX 460"),
("Volkswagen", "Passat"),
("Land Rover", "Range Rover Sport"),
("Nissan", "Versa"),
("Volvo", "XC40"),
("Mercedes-Benz", "E 350"),
("Chrysler", "300"),
("Chevrolet", "Impala"),
("Subaru", "Legacy"),
("Acura", "TLX"),
("Mercedes-Benz", "Sprinter"),
("Cadillac", "CT5"),
("Mercedes-Benz", "GLA 250"),
("Hyundai", "Santa Cruz"),
("Tesla", "Model S"),
("Mercedes-Benz", "GLB 250"),
("INFINITI", "Q50"),
("Kia", "K5"),
("Cadillac", "XT6"),
("Audi", "Q3"),
("INFINITI", "QX50"),
("Ford", "Transit 250"),
("Ford", "Mustang Mach-E"),
("Kia", "Seltos"),
("MAZDA", "MX-5 Miata"),
("Audi", "A5"),
("Lincoln", "Aviator"),
("BMW", "X1"),
("Kia", "Rio"),
("Chevrolet", "Express 2500"),
("Ford", "Transit 350"),
("Toyota", "Venza"),
("Mercedes-Benz", "S 500"),
("Cadillac", "Escalade ESV"),
("Jeep", "Wagoneer"),
("Chevrolet", "Bolt"),
("MINI", "Cooper Countryman"),
("Toyota", "Sequoia"),
("Mercedes-Benz", "CLA 250"),
("BMW", "X7"),
("Cadillac", "CTS"),
("Hyundai", "Venue"),
("Volkswagen", "ID.4"),
("Toyota", "Avalon"),
("Jeep", "Patriot"),
("Tesla", "Model Y"),
("Nissan", "Leaf"),
("Audi", "A3"),
("Acura", "Integra"),
("Ford", "Transit Connect"),
("Lexus", "NX 300"),
("Audi", "A6"),
("Mercedes-Benz", "EQS 450+"),
("Chevrolet", "Spark"),
("Jaguar", "F-PACE"),
("Mercedes-Benz", "S 580"),
("Chevrolet", "Sonic"),
("Lincoln", "Navigator"),
("Toyota", "C-HR"),
("Ford", "Fiesta"),
("RAM", "ProMaster City"),
("Volvo", "S60"),
("BMW", "330i xDrive"),
("Ford", "Flex"),
("MAZDA", "MAZDA6"),
("Toyota", "Corolla Cross"),
("Lincoln", "MKZ"),
("Chevrolet", "Express 3500"),
("Hyundai", "Accent"),
("Land Rover", "Discovery Sport"),
("Tesla", "Model X"),
("Honda", "Fit"),
("Alfa Romeo", "Stelvio"),
("Chrysler", "200"),
("Volkswagen", "Beetle"),
("Cadillac", "CT4"),
("Ford", "Maverick"),
("Volkswagen", "GTI"),
("Lincoln", "MKC"),
("Porsche", "Panamera"),
("Ford", "F450"),
("Lexus", "NX 350"),
("Chrysler", "Town & Country"),
("Kia", "Stinger"),
("Land Rover", "Range Rover Velar"),
("Audi", "S5"),
("BMW", "330i"),
("Volkswagen", "Golf"),
("Mercedes-Benz", "GLS 450"),
("Lexus", "IS 350"),
("Land Rover", "Range Rover Evoque"),
("Toyota", "Prius Prime"),
("Acura", "ILX"),
("Genesis", "G70"),
("Ford", "Taurus"),
("Hyundai", "Veloster"),
("Lexus", "IS 300"),
("Land Rover", "Defender"),
("Genesis", "GV80"),
("Alfa Romeo", "Giulia"),
("BMW", "X6"),
("Hyundai", "Ioniq 5"),
("Audi", "SQ5"),
("BMW", "328i"),
("BMW", "i3"),
("Cadillac", "ATS"),
("Mercedes-Benz", "S 550"),
("Lincoln", "Navigator L"),
("Mercedes-Benz", "E 450"),
("Buick", "LaCrosse"),
("Ford", "E-350 and Econoline 350"),
("BMW", "M3"),
("Mercedes-Benz", "GLE 53 AMG"),
("Lexus", "IS 250"),
("Mercedes-Benz", "E 300"),
("Cadillac", "SRX"),
("GMC", "Savana 2500"),
("INFINITI", "QX55"),
("Mitsubishi", "Eclipse Cross"),
("Audi", "Q8"),
("INFINITI", "Q60"),
("Kia", "Sedona"),
("Lincoln", "MKX"),
("Audi", "e-tron"),
("Chevrolet", "Volt"),
("BMW", "X4"),
("Chevrolet", "Bolt EUV"),
("Volvo", "C40"),
("Maserati", "Ghibli"),
("Lexus", "ES 300h"),
("Jaguar", "F-TYPE"),
("Cadillac", "XTS"),
("Genesis", "GV70"),
("BMW", "430i xDrive"),
("BMW", "430i"),
("BMW", "Z4"),
("BMW", "M4"),
("Land Rover", "Discovery"),
("Lexus", "GS 350"),
("Mercedes-Benz", "A 220"),
("Dodge", "Ram 1500 Truck"),
("Ford", "F550"),
("Hyundai", "Ioniq"),
("Mercedes-Benz", "ML 350"),
("Genesis", "G80"),
("MINI", "Cooper Clubman"),
("Maserati", "Levante"),
("Mercedes-Benz", "AMG GT"),
("BMW", "530i xDrive"),
("Lincoln", "Continental"),
("Chrysler", "Voyager"),
("Lexus", "LS 460"),
("MAZDA", "MX-5 Miata RF"),
("FIAT", "500"),
("Cadillac", "CT6"),
("MAZDA", "CX-3"),
("BMW", "M5"),
("BMW", "328i xDrive"),
("Hyundai", "Genesis"),
("Kia", "EV6"),
("INFINITI", "G37"),
("Audi", "A8"),
("Audi", "S4"),
("BMW", "X2"),
("BMW", "530i"),
("Lexus", "UX 250h"),
("Lexus", "RX 350L"),
("Mercedes-Benz", "G 63 AMG"),
("Nissan", "Juke"),
("Volkswagen", "Arteon"),
("Honda", "Insight"),
("Lexus", "RC 350"),
("RAM", "5500"),
("Audi", "A7"),
("Lexus", "NX 200t"),
("Nissan", "370Z"),
("Porsche", "Boxster"),
("BMW", "540i"),
("Buick", "Regal"),
("Dodge", "Dart"),
("BMW", "540i xDrive"),
("Mercedes-Benz", "GLE 450"),
("Ford", "Expedition EL"),
("Jeep", "Grand Wagoneer"),
("Bentley", "Continental"),
("Dodge", "Ram 2500 Truck"),
("Jeep", "Liberty"),
("Kia", "Carnival"),
("Mitsubishi", "Mirage G4"),
("Mercedes-Benz", "GL 450"),
("Mitsubishi", "Mirage"),
("Lexus", "RX 450h"),
("Porsche", "Taycan"),
("Acura", "TL"),
("Lexus", "CT 200h"),
("Nissan", "NV"),
("BMW", "440i xDrive"),
("Mercedes-Benz", "C 43 AMG"),
("Mercedes-Benz", "EQS 580"),
("Toyota", "Supra"),
("Mercedes-Benz", "GLK 350"),
("Lexus", "LS 500"),
("Toyota", "Prius C"),
("Toyota", "Yaris"),
("Jaguar", "XF"),
("Nissan", "Versa Note"),
("BMW", "335i"),
("Nissan", "Xterra"),
("Lexus", "NX 250"),
("Toyota", "FJ Cruiser"),
("Audi", "RS 5"),
("Volvo", "V60"),
("Audi", "S3"),
("BMW", "740i"),
("BMW", "128i"),
("Buick", "Verano"),
("Subaru", "BRZ"),
("Audi", "Q4 e-tron"),
("Chevrolet", "Avalanche"),
("Mercedes-Benz", "SL 550"),
("Ford", "C-MAX"),
("Toyota", "GR86"),
("BMW", "750i xDrive"),
("Ford", "Transit 150"),
("Mercedes-Benz", "Metris"),
("Mercedes-Benz", "S 560"),
("Nissan", "NV200"),
("Volkswagen", "Golf R"),
("Mercedes-Benz", "SL 63 AMG"),
("BMW", "M850i xDrive"),
("Lexus", "LX 570"),
("Mercedes-Benz", "G 550"),
("Ford", "E-450 and Econoline 450"),
("Ford", "E-Transit"),
("Mercedes-Benz", "C 250"),
("Mercedes-Benz", "CLS 450"),
("Mercedes-Benz", "S 63 AMG"),
("BMW", "530e"),
("BMW", "428i"),
("Mercedes-Benz", "GLC 43 AMG"),
("Volvo", "S90"),
("Dodge", "Avenger"),
("Lexus", "NX 300h"),
("Mercedes-Benz", "GLE 43 AMG"),
("Mercedes-Benz", "E 400"),
("Toyota", "Prius V"),
("BMW", "X5 M"),
("GMC", "Savana 3500"),
("Scion", "tC"),
("Volkswagen", "CC"),
("Acura", "TSX"),
("BMW", "228i xDrive"),
("BMW", "535i xDrive"),
("Porsche", "Cayman"),
("Subaru", "Impreza WRX"),
("BMW", "535i"),
("BMW", "M8"),
("Bentley", "Bentayga"),
("Maserati", "Quattroporte"),
("BMW", "M550i xDrive"),
("Jaguar", "XE"),
("Hyundai", "Kona N"),
("Porsche", "718 Cayman"),
("BMW", "M2"),
("Mercedes-Benz", "C 63 AMG"),
("BMW", "M340i"),
("Hyundai", "Elantra N"),
("BMW", "528i"),
("Ford", "E-250 and Econoline 250"),
("BMW", "i4"),
("FIAT", "500X"),
("BMW", "iX"),
("Audi", "TT"),
("Lexus", "IS 200t"),
("Maserati", "GranTurismo"),
("Dodge", "Ram 3500 Truck"),
("BMW", "650i"),
("Lexus", "UX 200"),
("Dodge", "Dakota"),
("INFINITI", "QX30"),
("Mercedes-Benz", "GLE 63 AMG"),
("Volkswagen", "Touareg"),
("Volkswagen", "e-Golf"),
("Lamborghini", "Huracan"),
("Lexus", "LC 500"),
("Land Rover", "LR4"),
("Lexus", "NX 350h"),
("BMW", "428i xDrive"),
("Jaguar", "XJ"),
("Lexus", "RC 300"),
("Toyota", "Mirai"),
("BMW", "330e"),
("Genesis", "G90"),
("Jaguar", "E-PACE"),
("Lamborghini", "Urus"),
("BMW", "M340i xDrive"),
("Audi", "RS 7"),
("Lexus", "ES 250"),
("Mercedes-Benz", "SL 55 AMG"),
("BMW", "320i"),
("Toyota", "Land Cruiser"),
("Ford", "Thunderbird"),
("Honda", "Element"),
("Scion", "xB"),
("BMW", "530e xDrive"),
("Porsche", "718 Boxster"),
("Buick", "Lucerne"),
("Mercedes-Benz", "E 53 AMG"),
("Mitsubishi", "Lancer"),
("Polestar", "Polestar 2"),
("RAM", "4500"),
("Scion", "FR-S"),
("Mercedes-Benz", "E 550"),
("Nissan", "GT-R"),
("BMW", "X6 M"),
("INFINITI", "Q70"),
("Audi", "R8"),
("Honda", "Clarity"),
("Mercedes-Benz", "E 63 AMG"),
("BMW", "320i xDrive"),
("Ford", "E-150 and Econoline 150"),
("Lexus", "GX 470"),
("Lincoln", "MKS"),
("BMW", "135i"),
("Mercedes-Benz", "GL 550"),
("Toyota", "86"),
("smart", "fortwo"),
("Chevrolet", "Express 1500"),
("BMW", "528i xDrive"),
("BMW", "M440i"),
("BMW", "230i"),
("INFINITI", "G35"),
("Mercedes-Benz", "S 450"),
("Mercedes-Benz", "SL 500"),
("BMW", "435i xDrive"),
("FIAT", "124 Spider"),
("Mercedes-Benz", "CLS 550"),
("Mercedes-Benz", "EQE 350+"),
("Mercury", "Grand Marquis"),
("Volkswagen", "Eos"),
("Chrysler", "PT Cruiser"),
("Lexus", "SC 430"),
("Lincoln", "Town Car"),
("Nissan", "Quest"),
("Audi", "S8"),
("BMW", "435i"),
("HUMMER", "H2"),
("Kia", "Cadenza"),
("BMW", "228i"),
("Chrysler", "Sebring"),
("Volvo", "XC70"),
("BMW", "335i xDrive"),
("Chevrolet", "Captiva Sport"),
("Ferrari", "California"),
("Ford", "Excursion"),
("BMW", "440i"),
("Chevrolet", "HHR"),
("INFINITI", "QX56"),
("INFINITI", "QX70"),
("MAZDA", "MAZDA5"),
("Pontiac", "G6"),
("Chevrolet", "Cobalt"),
("Rivian", "R1T"),
("Audi", "S6"),
("BMW", "750i"),
("BMW", "M240i xDrive"),
("BMW", "i8"),
)
MAKE_MODEL_TO_INDEX = {x: i for i, x in enumerate(MAKES_MODELS)}
NUM_MAKE_MODELS = len(MAKE_MODEL_TO_INDEX) + 1