Apartment Prices in Bogotá

In this project I am going to make a model to predict the apartment prices in Bogotá using the size of the apartment.

First I am going to explore the data.

Import the libraries

import pandas as pd import matplotlib.pyplot as plt import plotly.express as px from sklearn.linear_model import LinearRegression from sklearn.metrics import mean_absolute_error from sklearn.model_selection import train_test_split

Read the data.

The data was obtained from the Properarti web page.

https://www.properati.com.ec/data/assets/img/properati-data/flag-col.png

df = pd.read_csv("/work/co_properties.csv")

It is necessary to use the properties that are apartments, with a surface covered major than 0, those must be located in Bogotá, the currency used is COP, the operation type is Venta and the price is different from zero. The duplicates are erased.

dfs=df[df["surface_covered"]>0] dfs=dfs[dfs["l3"]=="Bogotá D.C"] apartamento=["Apartamento"] dfs = dfs[dfs["property_type"].isin(apartamento)] dfs = dfs[dfs["currency"]=="COP"] dfs = dfs[dfs["operation_type"]=="Venta"] dfs = dfs[["property_type","title","lat","lon","l4","price","surface_covered"] ] dfs = dfs.rename(index=str, columns={"l4":"neighborhood"}) dfs=dfs[dfs["price"]!=0] dfs.drop_duplicates(inplace=True) #dfs.dropna(inplace=True) dfs.reset_index(drop=True, inplace=True)

The data frame head.

dfs.head()

Explore

plt.boxplot(dfs["surface_covered"]);

In the surfaced covered box plot is obvious that there are many outliers. It is necessary to remove them.

q10=dfs["surface_covered"].quantile(0.1) q90=dfs["surface_covered"].quantile(0.9) mask=(dfs["surface_covered"]<q90) & (dfs["surface_covered"]>q10) dfs=dfs[mask]

With a histogram of the surface covered we can analyze the Distribution of the apartment sizes.

plt.hist(dfs["surface_covered"]) plt.xlabel("Area [sq meters]") plt.title("Distribution of Apartment Sizes");

With a scatter plot price versus area it is possible to analyze if there is a correlation between Area and Price.

plt.scatter(x=dfs["surface_covered"], y=dfs["price"]) plt.xlabel("Area [sq meters]") plt.ylabel("Price [COP]"); plt.title("Bogotá: Price vs. Area");

There is a correlation between price and area according with the graphic. The correlation coefficient is 0.8. It is possible to say that there is a strong correlation between the two variables.

dfs["price"].corr(dfs["surface_covered"])

Model

First the data is separated in the feature X, surfaced covered, and the target y, price.

features = ["surface_covered"] X = dfs[features] X.shape

target = "price" y = dfs[target] y.shape

Then the X and y is split in X train, y train and X test and y test.

X_train, X_test, y_train, y_test = train_test_split( X, y, test_size=0.2, random_state=42 )

Build the model

First the Baseline is calculated.

y_mean = y_train.mean() y_pred_baseline = [y_mean]*len(y_train)

The y pred baseline is drawing in the graphic of Price vs Area.

plt.plot(X_train, y_pred_baseline, color="orange", label="Baseline Model") plt.scatter(X_train, y_train) plt.xlabel("Area [sq meters]") plt.ylabel("Price [USD]") plt.title("Buenos Aires: Price vs. Area") plt.legend();

Now it is calculated the mean absolute error of the baseline.

mae_baseline = mean_absolute_error(y_pred_baseline, y_train) print("Baseline MAE:", round(mae_baseline, 2))

Iterate

It is used Linear Regression. We instantiate it and fit it.

model=LinearRegression()

model.fit(X_train, y_train)

Evaluate

First we predict with the X train.

y_pred_training=model.predict(X_train)

It is calculated the mean absolute error of training.

mae_training=mean_absolute_error(y_train, y_pred_training) print("Training MAE:", round(mae_training, 2))

mae_baseline-mae_training

We can see the model beats the baseline in 175066576.

Finally, we evaluate the model with the test data.

y_pred_test = pd.Series(model.predict(X_test)) y_pred_test.head()

It is calculated the mean absolute error of test data.

mae_test=mean_absolute_error(y_test, y_pred_test) print("Test MAE:", round(mae_test, 2))

mae_baseline-mae_test

We can see the model beats the baseline in 184460075.

Communicate Results

Finally the results are communicated. We obtain the model intercept and coefficient.

intercept = round(model.intercept_, 2) print("Model Intercept:", intercept)

coefficient = round(model.coef_[0],2) print('Model coefficient for "surface_covered_in_m2":', coefficient)

The next is the formula for the apartment price.

print(f"apt_price = {intercept} + {coefficient} * surface_covered")

We plot the graphic of the formula in the plot of Price vs Surface.

plt.plot(X_train, model.predict(X_train), color="red", label="Linear Model") plt.scatter(X_train, y_train) plt.xlabel("surface covered [sq meters]") plt.ylabel("price [usd]") plt.legend();

We can see that our model is capable to predict the prices of the apartments in Bogotá.