fivedollarwitch commited on
Commit
b08dc53
·
verified ·
1 Parent(s): 6936c1b

Upload 4 files

Browse files
MLBaseModel.py ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import torch
2
+ import torch.nn as nn
3
+
4
+ class MLBaseModel(nn.Module):
5
+ def __init__(self, input_dim):
6
+ super(MLBaseModel, self).__init__()
7
+ self.fc1 = nn.Linear(input_dim, 256)
8
+ self.fc2 = nn.Linear(256, 128)
9
+ self.fc3 = nn.Linear(128, 64)
10
+ self.fc4 = nn.Linear(64, 2) # Output 2 values: sale_price and days_on_market
11
+
12
+ def forward(self, x):
13
+ x = torch.relu(self.fc1(x))
14
+ x = torch.relu(self.fc2(x))
15
+ x = torch.relu(self.fc3(x))
16
+ x = self.fc4(x)
17
+ return x
model_files/label_encoder.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2fd02a7b03d8323c065fbb3a010f92710b96d568fe18f48922de1b106d0928e9
3
+ size 306
model_files/real_estate_model.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:39811330ff68ab47c4838bceacc0fb8e87c0120282506d03f1119992996062a0
3
+ size 173496
model_files/scaler.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7c321473fc77b93773052027f5dc5e2ae03c51b3b9a39c093c674444103caabc
3
+ size 600