Increase dropout on regression model to cut down on overfitting
This commit is contained in:
@@ -14,8 +14,8 @@ import sys
|
||||
NUM_CLASSES = 3
|
||||
EMBEDDING_MODEL = "all-mpnet-base-v2"
|
||||
HIDDEN_DIM = 256
|
||||
DROPOUT = 0.3
|
||||
LEARNING_RATE = 1e-3
|
||||
DROPOUT = 0.4
|
||||
LEARNING_RATE = 2e-3
|
||||
WEIGHT_DECAY = 1e-4
|
||||
BATCH_SIZE = 64
|
||||
NUM_EPOCHS = 30
|
||||
@@ -90,8 +90,6 @@ class LogisticNet(nn.Module):
|
||||
return self.net(x)
|
||||
|
||||
|
||||
# ── Metrics ───────────────────────────────────────────────────────────────────
|
||||
|
||||
def evaluate(model, loader, device):
|
||||
model.eval()
|
||||
all_preds, all_labels = [], []
|
||||
|
||||
@@ -64,7 +64,7 @@ def render():
|
||||
if ("flan" in file_path.name):
|
||||
thresh = 0.94
|
||||
if ("regression" in file_path.name):
|
||||
thresh = 0.7
|
||||
thresh = 0.75
|
||||
|
||||
st.subheader(f"File: {file_path.name}")
|
||||
|
||||
|
||||
Reference in New Issue
Block a user