Fix #199

konlpy · Jul 31, 2018 · 91a181d · 91a181d
1 parent c1dbc2a
commit 91a181d
Show file tree

Hide file tree

Showing 5 changed files with 10 additions and 10 deletions.
diff --git a/konlpy/jvm.py b/konlpy/jvm.py
@@ -12,7 +12,7 @@
 from konlpy import utils
 
 
-def init_jvm(jvmpath=None):
+def init_jvm(jvmpath=None, max_heap_size=1024):
     """Initializes the Java virtual machine (JVM).
 
     :param jvmpath: The path of the JVM. If left empty, inferred by :py:func:`jpype.getDefaultJVMPath`.
@@ -62,6 +62,6 @@ def init_jvm(jvmpath=None):
     if jvmpath:
         jpype.startJVM(jvmpath, '-Djava.class.path=%s' % classpath,
                                 '-Dfile.encoding=UTF8',
-                                '-ea', '-Xmx1024m')
+                                '-ea', '-Xmx{}m'.format(max_heap_size))
     else:
         raise ValueError("Please specify the JVM path.")
diff --git a/konlpy/tag/_hannanum.py b/konlpy/tag/_hannanum.py
@@ -102,9 +102,9 @@ def morphs(self, phrase):
 
         return [s for s, t in self.pos(phrase)]
 
-    def __init__(self, jvmpath=None):
+    def __init__(self, jvmpath=None, max_heap_size=1024):
         if not jpype.isJVMStarted():
-            jvm.init_jvm(jvmpath)
+            jvm.init_jvm(jvmpath, max_heap_size)
 
         jhannanumJavaPackage = jpype.JPackage('kr.lucypark.jhannanum.comm')
         HannanumInterfaceJavaClass = jhannanumJavaPackage.HannanumInterface

diff --git a/konlpy/tag/_kkma.py b/konlpy/tag/_kkma.py
@@ -89,9 +89,9 @@ def sentences(self, phrase):
         if not sentences: return []
         return [sentences.get(i).getSentence() for i in range(sentences.size())]
 
-    def __init__(self, jvmpath=None):
+    def __init__(self, jvmpath=None, max_heap_size=1024):
         if not jpype.isJVMStarted():
-            jvm.init_jvm(jvmpath)
+            jvm.init_jvm(jvmpath, max_heap_size)
 
         kkmaJavaPackage = jpype.JPackage('kr.lucypark.kkma')
         KkmaInterfaceJavaClass = kkmaJavaPackage.KkmaInterface

diff --git a/konlpy/tag/_komoran.py b/konlpy/tag/_komoran.py
@@ -85,9 +85,9 @@ def morphs(self, phrase):
 
         return [s for s, t in self.pos(phrase)]
 
-    def __init__(self, jvmpath=None, userdic=None, modelpath=None):
+    def __init__(self, jvmpath=None, userdic=None, modelpath=None, max_heap_size=1024):
         if not jpype.isJVMStarted():
-            jvm.init_jvm(jvmpath)
+            jvm.init_jvm(jvmpath, max_heap_size)
 
         if modelpath:
             self.modelpath = modelpath

diff --git a/konlpy/tag/_okt.py b/konlpy/tag/_okt.py
@@ -81,9 +81,9 @@ def phrases(self, phrase):
 
         return [p for p in self.jki.phrases(phrase).toArray()]
 
-    def __init__(self, jvmpath=None):
+    def __init__(self, jvmpath=None, max_heap_size=1024):
         if not jpype.isJVMStarted():
-            jvm.init_jvm(jvmpath)
+            jvm.init_jvm(jvmpath, max_heap_size)
 
         oktJavaPackage = jpype.JPackage('kr.lucypark.okt')
         OktInterfaceJavaClass = oktJavaPackage.OktInterface