aur-beau commited on
Commit
87bc31c
·
1 Parent(s): d329a6c

removed comments

Browse files
Files changed (1) hide show
  1. tasks/text.py +0 -6
tasks/text.py CHANGED
@@ -74,9 +74,6 @@ async def evaluate_text(request: TextEvaluationRequest):
74
  # Make random predictions (placeholder for actual model inference)
75
  true_labels = test_dataset["label"]
76
 
77
- # load the xgboost model
78
- # with open("models/stella_400_xgb_500.pkl",'rb') as f:
79
- # xgbclassifier = pickle.load(f)
80
  from torch import nn, optim
81
 
82
  class SimpleNN2(nn.Module):
@@ -103,9 +100,6 @@ async def evaluate_text(request: TextEvaluationRequest):
103
  # Set the model to evaluation mode
104
  model_nn.eval()
105
 
106
- # make inference
107
- # predictions = xgbclassifier.predict(embeddings)
108
-
109
  # Make predictions
110
  with torch.no_grad():
111
  outputs = model_nn(embeddings)
 
74
  # Make random predictions (placeholder for actual model inference)
75
  true_labels = test_dataset["label"]
76
 
 
 
 
77
  from torch import nn, optim
78
 
79
  class SimpleNN2(nn.Module):
 
100
  # Set the model to evaluation mode
101
  model_nn.eval()
102
 
 
 
 
103
  # Make predictions
104
  with torch.no_grad():
105
  outputs = model_nn(embeddings)