migrating towards openAI gym's API

Author: thoward27

source/AlphaCompile.py

@@ -3,17 +3,18 @@
 
 from PyTorchRL.agents.AlphaZero import AlphaZero
 from source.config import FLAGS
+from source.programs import Programs
 
 STEPS = 10
 
 
 class AlphaCompile(AlphaZero):
     def __init__(self, input_dim: int = 5, output_dim: int = len(FLAGS) + 1):
-        hidden = 5
+        self.hidden = 5
         body = nn.Sequential(
-            nn.Linear(input_dim, hidden),
-            nn.Dropout(0.5),
-            nn.Linear(hidden, hidden),
+            nn.Linear(input_dim, self.hidden),
+            nn.Dropout(0.4),
+            nn.Linear(self.hidden, self.hidden),
         )
         super().__init__(input_dim, output_dim, body)
 
@@ -30,4 +31,6 @@ def train(programs):
 
 if __name__ == "__main__":
     log_param("steps", STEPS)
-    train([])
+    programs = Programs()
+    programs = programs.filter(programs[0])
+

source/_ck.py

@@ -0,0 +1,4 @@
+
+import ck
+
+results = ck.compile

source/config.py

@@ -1,14 +1,7 @@
 """ Settings file.
 """
-import os
-import sqlite3 as sql
-from datetime import datetime
-from decimal import Decimal
 from enum import Enum
 
-sql.register_adapter(Decimal, lambda d: str(d))
-sql.register_converter("DEC", lambda s: Decimal(s.decode('utf-8')))
-
 
 class Features(Enum):
     HYBRID = 0
@@ -17,10 +10,6 @@ class Features(Enum):
 
 
 EPOCHS = 100
-THREADED_RUNTIMES = False
-USE_RUNTIMES = True
-NOW = datetime.now().strftime("%Y_%m_%d_%H_%M_%S")
-RUN_DIR = os.path.abspath('./runs/run_{}'.format(NOW))
 
 # noinspection SpellCheckingInspection
 FLAGS = [
@@ -34,44 +23,3 @@ class Features(Enum):
 ]
 N_FLAGS = len(FLAGS)
 
-ACTIONS = [(
-    '-O3',
-    *(f for f, s in zip(FLAGS, list(format(a, '0%sb' % N_FLAGS))) if s == '1')
-) for a in range(2 ** N_FLAGS)]
-
-N_ACTIONS = len(ACTIONS)
-
-# noinspection SpellCheckingInspection
-LOG_CONFIG = {
-    'version': 1,
-    'disable_existing_loggers': False,
-    'formatters': {
-        'simple': {
-            'format': '%(levelname)s: %(filename)s: %(message)s',
-        },
-        'detailed': {
-            'class': 'logging.Formatter',
-            'format': '%(asctime)s, %(levelname)-6s, %(filename)-6s, %(funcName)s, %(message)s',
-        }
-    },
-    'handlers': {
-        'console': {
-            'class': 'logging.StreamHandler',
-            'level': 'INFO',
-            'formatter': 'simple',
-        },
-        'event_file': {
-            'class': 'logging.FileHandler',
-            'level': 'INFO',
-            'formatter': 'detailed',
-            'filename': '{}/events.log'.format(RUN_DIR),
-        },
-    },
-    'loggers': {
-        '': {
-            'handlers': ['console', 'event_file'],
-            'level': 'DEBUG',
-            'propogate': False,
-        }
-    }
-}

source/metrics.py

@@ -6,7 +6,7 @@
 
 import source.config as c
 import source.utils as u
-from source.programs import Program
+from Benchmarks.program import Program
 
 
 class Metric: