32 lines
578 B
Python
32 lines
578 B
Python
|
import pandas as pd
|
||
|
|
|
||
|
|
from constants import DATA_SIZE
|
||
|
|
|
||
|
|
|
||
|
|
def load_data():
|
||
|
|
data = read_data()
|
||
|
|
Y = data['Model']
|
||
|
|
X = data[['Year', 'Price', 'Mileage']]
|
||
|
|
|
||
|
|
return X, Y
|
||
|
|
|
||
|
|
|
||
|
|
def fit_model(model, x):
|
||
|
|
transformed = model.fit_transform(x)
|
||
|
|
|
||
|
|
return transformed[:, 0], transformed[:, 1]
|
||
|
|
|
||
|
|
|
||
|
|
def create_data():
|
||
|
|
data = read_data()
|
||
|
|
return data[['Year', 'Price']]
|
||
|
|
|
||
|
|
|
||
|
|
def read_data():
|
||
|
|
data = pd.read_csv('true_car_listings.csv')[:DATA_SIZE]
|
||
|
|
|
||
|
|
unique_numbers = list(set(data['Model']))
|
||
|
|
data['Model'] = data['Model'].apply(unique_numbers.index)
|
||
|
|
return data
|
||
|
|
|