gnh1201/lotto_genetic_algorithm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update main.py

Browse Source
This commit is contained in:
Namhyeon Go 2021-04-04 11:03:20 +09:00 committed by GitHub
parent 20881eefb2
commit 4fe896350a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
main.py
View File

@ -9,7 +9,7 @@ from geneticalgorithm2 import geneticalgorithm2 as ga
cols = [1, 13, 14, 15, 16, 17, 18]
df = pd.read_excel('excel.xlsx', skiprows=2, usecols=cols, names=[0, 1, 2, 3, 4, 5, 6])
rows = df.values[:200]
rows = df.values[:10]
def make_num(x, n, a, b, c, min, max):
seed = int(x*a + n*b + c)
@ -36,8 +36,15 @@ def f(X):
return -score
varbound = np.array([[0,10000]]*3)
varbound = np.array([[0, 100000]]*3)
model = ga(function=f, dimension=3, variable_type='int', variable_boundaries=varbound)
model.run()
solution = model.output_dict
variable = model.output_dict['variable']
_x = 958 # prediction
print ('Recommended numbers:')
print (', '.join(str(make_num(_x, n, variable[0], variable[1], variable[2], 1, 45)) for n in range(1, 7)))