pyai-new 0.1.0__tar.gz

pyai_new-0.1.0/PKG-INFO

@@ -0,0 +1,12 @@
+Metadata-Version: 2.4
+Name: pyai-new
+Version: 0.1.0
+Summary: Мой собственный движок нейросетей на C++
+Author: Глеб
+Classifier: Programming Language :: Python :: 3
+Classifier: Operating System :: Microsoft :: Windows
+Requires-Python: >=3.6
+Dynamic: author
+Dynamic: classifier
+Dynamic: requires-python
+Dynamic: summary

pyai_new-0.1.0/README.md

@@ -0,0 +1,13 @@
+# PyAI
+Мой личный движок для обучения нейросетей, написанный на C++.
+
+## Установка
+pip install .
+
+## Пример использования
+from pyai import PyAI
+
+ai = PyAI()
+ai.create(2, 1, [16, 16])
+ai.train([0.5, 0.5], [0.5])
+print(ai.predict([0.5, 0.5]))

pyai_new-0.1.0/pyai/__init__.py

@@ -0,0 +1 @@
+from .engine import PyAI

pyai_new-0.1.0/pyai/engine.py

@@ -0,0 +1,52 @@
+import ctypes
+import os
+
+class PyAI:
+    def __init__(self, lib_path=None):
+        if lib_path is None:
+            # Ищем DLL в той же папке, где лежит этот файл
+            lib_path = os.path.join(os.path.dirname(__file__), "pyai.dll")
+            
+        self.lib = ctypes.CDLL(lib_path)
+        
+        self.lib.create_network_from_config.argtypes = [
+            ctypes.c_int, ctypes.c_int, ctypes.POINTER(ctypes.c_int), ctypes.c_int, ctypes.c_float
+        ]
+        self.lib.create_network_from_config.restype = ctypes.c_int
+        
+        self.lib.forward_pass.argtypes = [
+            ctypes.c_int, ctypes.POINTER(ctypes.c_float), ctypes.POINTER(ctypes.c_float)
+        ]
+        
+        self.lib.train_step.argtypes = [
+            ctypes.c_int, ctypes.POINTER(ctypes.c_float), ctypes.POINTER(ctypes.c_float)
+        ]
+        self.lib.train_step.restype = ctypes.c_float
+
+        self.net_id = -1
+        self.input_size = 0
+        self.output_size = 0
+
+    def create(self, INPUT_NEURONS, OUTPUT_NEURONS, HIDDEN_LAYERS, learning_rate=0.05):
+        self.input_size = INPUT_NEURONS
+        self.output_size = OUTPUT_NEURONS
+        
+        num_hidden = len(HIDDEN_LAYERS)
+        hidden_arr = (ctypes.c_int * num_hidden)(*HIDDEN_LAYERS)
+        
+        self.net_id = self.lib.create_network_from_config(
+            self.input_size, self.output_size, hidden_arr, num_hidden, learning_rate
+        )
+        return self.net_id
+
+    def predict(self, inputs):
+        in_arr = (ctypes.c_float * self.input_size)(*inputs)
+        out_arr = (ctypes.c_float * self.output_size)()
+        self.lib.forward_pass(self.net_id, in_arr, out_arr)
+        return list(out_arr)
+
+    def train(self, inputs, targets):
+        in_arr = (ctypes.c_float * self.input_size)(*inputs)
+        target_arr = (ctypes.c_float * self.output_size)(*targets)
+        loss = self.lib.train_step(self.net_id, in_arr, target_arr)
+        return loss

pyai_new-0.1.0/pyai_new.egg-info/PKG-INFO

@@ -0,0 +1,12 @@
+Metadata-Version: 2.4
+Name: pyai-new
+Version: 0.1.0
+Summary: Мой собственный движок нейросетей на C++
+Author: Глеб
+Classifier: Programming Language :: Python :: 3
+Classifier: Operating System :: Microsoft :: Windows
+Requires-Python: >=3.6
+Dynamic: author
+Dynamic: classifier
+Dynamic: requires-python
+Dynamic: summary

pyai_new-0.1.0/pyai_new.egg-info/SOURCES.txt

@@ -0,0 +1,9 @@
+README.md
+setup.py
+pyai/__init__.py
+pyai/engine.py
+pyai/pyai.dll
+pyai_new.egg-info/PKG-INFO
+pyai_new.egg-info/SOURCES.txt
+pyai_new.egg-info/dependency_links.txt
+pyai_new.egg-info/top_level.txt

pyai_new-0.1.0/pyai_new.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

pyai_new-0.1.0/pyai_new.egg-info/top_level.txt

@@ -0,0 +1 @@
+pyai

pyai_new-0.1.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

pyai_new-0.1.0/setup.py

@@ -0,0 +1,18 @@
+from setuptools import setup, find_packages
+
+setup(
+    name="pyai-new",  # <--- Вот тут имя нашего пакета!
+    version="0.1.0",
+    author="Глеб",
+    description="Мой собственный движок нейросетей на C++",
+    packages=find_packages(),
+    include_package_data=True,
+    package_data={
+        'pyai': ['*.dll'], # Упаковываем DLL внутрь пакета
+    },
+    classifiers=[
+        "Programming Language :: Python :: 3",
+        "Operating System :: Microsoft :: Windows",
+    ],
+    python_requires='>=3.6',
+)