Mercurial > repos > bgruening > tabpfn

changeset 3:4a92db686946 draft

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/tabpfn commit c1c3b6d5abd35875890c45baacab073b5e749537
author bgruening
date Mon, 20 Jan 2025 15:45:07 +0000
parents abe1c3ac9145
children 3957cd124013
files main.py
diffstat 1 files changed, 2 insertions(+), 2 deletions(-) [+]
line wrap: on
line diff
--- a/main.py	Fri Jan 17 22:23:21 2025 +0000
+++ b/main.py	Mon Jan 20 15:45:07 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: