Skip to content

Commit e866171

Browse files
committed
model transfer
1 parent 806b117 commit e866171

File tree

1 file changed

+11
-8
lines changed

1 file changed

+11
-8
lines changed

src/baskerville/util/model_transfer.py

Lines changed: 11 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -75,20 +75,23 @@ def model_transfer(
7575
for model in models_in:
7676
print(f'Getting model with id: {model.id}')
7777
model_out = Model()
78-
model_out.scaler = model.scaler
79-
model_out.classifier = model.classifier
8078
model_out.features = model.features
8179
model_out.algorithm = model.algorithm
82-
model_out.analysis_notebook = model.analysis_notebook
83-
model_out.created_at = model.created_at
80+
model_out.scaler_type = model.scaler_type
81+
model_out.parameters = model.parameters
82+
model_out.recall = model.recall
83+
model_out.precision = model.precision
8484
model_out.f1_score = model.f1_score
85+
model_out.classifier = model.classifier
86+
model_out.scaler = model.scaler
87+
model_out.host_encoder = model.host_encoder
8588
model_out.n_training = model.n_training
8689
model_out.n_testing = model.n_testing
90+
model_out.analysis_notebook = model.analysis_notebook
8791
model_out.notes = model.notes
88-
model_out.parameters = model.parameters
89-
model_out.precision = model.precision
90-
model_out.recall = model.recall
91-
# model_out.request_sets = model.request_sets
92+
model_out.threshold = model.threshold
93+
model_out.request_sets = None
94+
model_out.created_at = model.created_at
9295
out_session.add(model_out)
9396
out_session.commit()
9497
except Exception:

0 commit comments

Comments
 (0)