move model.py to model folder

james397520 · Oct 17, 2023 · 1e335dd · 1e335dd
1 parent bb32464
commit 1e335dd
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/inference.py b/inference.py
@@ -4,7 +4,7 @@
 import torch.nn as nn
 import torch.nn.functional as F
 from torch.utils.data import Dataset, DataLoader
-from model import HousePriceModel, TransformerRegressor
+from model.model import HousePriceModel, TransformerRegressor
 from dataloader import  HousePriceTestDataset, min_max_denormalize, z_score_denormalize
 import platform
 

diff --git a/model.py → model/model.py b/model.py → model/model.py
diff --git a/train.py b/train.py
@@ -2,7 +2,7 @@
 import pandas as pd
 from torch.utils.data import Dataset, DataLoader
 from sklearn.preprocessing import StandardScaler
-from model import HousePriceModel, TransformerRegressor
+from model.model import HousePriceModel, TransformerRegressor
 from dataloader import HousePriceTrainDataset
 import platform
 
@@ -20,7 +20,7 @@ def main():
     '移轉層次': 'min-max',
     '總樓層數': 'min-max',
     '屋齡': 'min-max',
-    '建物面積': 'min-max',
+    # '建物面積': 'min-max',
     '車位面積': 'min-max',
     '車位個數': 'min-max',
     '橫坐標': 'min-max', #z-score
@@ -44,7 +44,7 @@ def main():
     print(input_dim)
     # Initialize model
     # model = HousePriceModel(input_dim)
-    model = TransformerRegressor(input_dim, 4, 6)
+    model = TransformerRegressor(input_dim, 4, 11)
 
     if gpu:
         model = model.cuda()