| Previous changeset 2:c081e5e1d7ce (2025-01-17) Next changeset 4:e7b4afedc471 (2025-02-11) |
|
Commit message:
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/tabpfn commit c1c3b6d5abd35875890c45baacab073b5e749537 |
|
modified:
main.py |
| b |
| diff -r c081e5e1d7ce -r 33d53eb476fd main.py --- a/main.py Fri Jan 17 22:23:34 2025 +0000 +++ b/main.py Mon Jan 20 15:45:17 2025 +0000 |
| [ |
| @@ -61,7 +61,7 @@ te_labels = [] s_time = time.time() if args["selected_task"] == "Classification": - classifier = TabPFNClassifier(device="cpu") + classifier = TabPFNClassifier() classifier.fit(tr_features, tr_labels) y_eval = classifier.predict(te_features) pred_probas_test = classifier.predict_proba(te_features) @@ -81,7 +81,7 @@ "Precision", ) else: - regressor = TabPFNRegressor(device="cpu") + regressor = TabPFNRegressor() regressor.fit(tr_features, tr_labels) y_eval = regressor.predict(te_features) if len(te_labels) > 0: |