Files
PSO/setup.py
jung-geun 6e838ddfd5 23-10-20
조기 수렴 시 파티클 리셋 적용
모델의 초기화 수정 => 랜덤값은 문제가 많음
미니배치 초기화 시 자동 shuffle 적용
negative 파티클 특정 수치마다 초기화
2023-10-20 05:47:25 +09:00

38 lines
1.0 KiB
Python

from setuptools import setup, find_packages
import pso
VERSION = pso.__version__
setup(
name="pso2keras",
version=VERSION,
description="Particle Swarm Optimization on tensorflow package",
author="pieroot",
author_email="jgbong0306@gmail.com",
url="https://github.com/jung-geun/PSO",
install_requires=[
"tqdm",
"numpy",
"pandas",
"ipython",
"tensorflow<=2.11.1,>=2.8.0",
"keras<=2.11.1,>=2.8.0",
],
packages=find_packages(exclude=[]),
keywords=["pso", "tensorflow", "keras"],
python_requires=">=3.8",
package_data={},
zip_safe=False,
long_description=open("README.md", encoding="UTF8").read(),
long_description_content_type="text/markdown",
license="MIT",
classifiers=[
"License :: OSI Approved :: MIT License",
"Programming Language :: Python :: 3 :: Only",
"Programming Language :: Python :: 3.8",
"Programming Language :: Python :: 3.9",
"Programming Language :: Python :: 3.10",
],
)