working !!

Author: behrica

simpletransformers.clj

@@ -1,13 +1,10 @@
-(ns scicloj.ml.simpletransformers)
+(ns scicloj.ml.simpletransformers
+   (:require [libpython-clj2.python :refer [py.- py.] :as py]))
+
 
-(require '[libpython-clj2.require :as py-req]
-         '[libpython-clj2.python.io-redirect]
-         '[libpython-clj2.python :refer [py.- py.] :as py])
 
 
 
-(py/from-import simpletransformers.classification ClassificationModel)
-(py-req/require-python '[pandas :as pd])
 
 (def  train-data  [
                    ["Example sentence belonging to class 1" 1]
@@ -21,18 +18,24 @@
 
 
 (println "train...")
+(py/with-gil-stack-rc-context
+
+  (py/from-import simpletransformers.classification ClassificationModel)
+
+
 
-(let [
-      train-df (pd/DataFrame train-data)
-      eval-df (pd/DataFrame eval-data)
+  (let [
+        pd (py/import-module "pandas")
+        train-df (py. pd DataFrame train-data)
+        eval-df (py. pd DataFrame eval-data)
 
-      model (ClassificationModel "bert" "prajjwal1/bert-tiny" :use_cuda false :args
-                                 {:num_train_epochs 1
-                                  :use_multiprocessing false
-                                  :overwrite_output_dir true})
+        model (ClassificationModel "bert" "prajjwal1/bert-tiny" :use_cuda false :args
+                                   {:num_train_epochs 1
+                                    :use_multiprocessing false
+                                    :overwrite_output_dir true})
 
 
-      x  (py. model train_model train-df)]
-  (println x))
+        x  (py. model train_model train-df)]
+    (println x)))
 
 (println "finished train")