def adj_max_speed_func():
params.update({
'max_speed':
max(1, min(30, params['max_speed'] + adjs['delta_max_speed']))
})
max_speed_tag.set_text('Max Speed(px): {}'.format(params['max_speed']))
max_speed_tag.__build__()
pygame.time.wait(150)
def adj_min_dist_func():
params.update({
'min_dist':
max(20, min(50, adjs['delta_min_dist'] + params['min_dist']))
})
min_dist_tag.set_text('Minimum distance(px): {}'.format(
params['min_dist']))
min_dist_tag.__build__()
pygame.time.wait(150)
def adj_influence_func():
params.update({
'influence':
max(20, min(800, adjs['delta_influence'] + params['influence']))
})
influence_tag.set_text('Visual perception(px): {}'.format(
params['influence']))
influence_tag.__build__()
pygame.time.wait(150)
def adj_align_factor_func():
params.update({
'align_factor':
max(0, min(0.5, adjs['delta_align_factor'] + params['align_factor']))
})
align_factor_tag.set_text('Alignment Factor: {:.2f}'.format(
round(params['align_factor'], 2)))
align_factor_tag.__build__()
pygame.time.wait(150)
def adj_cohesion_factor_func():
params.update({
'cohesion_factor':
max(
0,
min(0.99,
adjs['delta_cohesion_factor'] + params['cohesion_factor']))
})
cohesion_factor_tag.set_text('Cohesion Factor: {:.2f}'.format(
round(params['cohesion_factor'], 2)))
cohesion_factor_tag.__build__()
pygame.time.wait(150)
def load_bert_char(vocab_file, params):
vocab = collections.OrderedDict()
index_vocab = collections.OrderedDict()
index = 0
with open(vocab_file, 'rb') as reader:
while True:
tmp = reader.readline()
token = convert_to_unicode(tmp)
if not token:
break
token = token.strip()
vocab[token] = index
index_vocab[index] = token
index += 1
if '<UNK' not in vocab:
length=len(vocab)
vocab.update({'<UNK>':length})
index_vocab.update({length:'<UNK>'})
params.update(vocab_size=len(vocab))
return vocab, index_vocab
def adj_noise_func():
params.update(
{'noise': max(0, min(30, adjs['delta_noise'] + params['noise']))})
noise_tag.set_text('Noise Bounds: +-{}'.format(params['noise']))
noise_tag.__build__()
pygame.time.wait(150)
def adj_jitter_func():
params.update({'jitter' : max(0, min(100, adjs['delta_jitter'] + params['jitter']))})
if params['jitter'] > 0: jitter_tag.set_text('Jitter (Radians): PI/{}'.format(params['jitter']))
else: jitter_tag.set_text('Jitter (Radians): OFF')
jitter_tag.__build__()
pygame.time.wait(150)
from config import window_size
from config import params
from pygameui.clickable.toggle_button import ToggleButton
from pygameui.text_object import TextObject
from pygameui.row_border import RowBorder
import pygame
always_limit_speed_tag = TextObject([0, 0], [255, 0, 0],
pygame.font.SysFont('arial', 20),
'Always Limit Speed: ')
toggle_always_limit_speed = ToggleButton(
[0, 0], [[255, 0, 0], [255, 0, 0]],
pygame.font.SysFont('arial', 20), ['ACTIVE', 'INACTIVE'], [30, 10],
on_click=lambda: params.update({'always_limit_speed': False}) if params[
'always_limit_speed'] else params.update({'always_limit_speed': True}))
always_limit_speed_row = RowBorder(
[window_size[0] // 4, 590], [0, 255, 255], 950, 10,
[always_limit_speed_tag, toggle_always_limit_speed])
estimator = tf.estimator.Estimator(model_fn=model_fn,
model_dir=params["model_dir"],
config=run_config)
if params['do_train']:
examples = pd.read_csv(params['calendar_data'])
x, y = Input_builder().create_RNN_input(
examples.iloc[:, -1].values,
train_window=params['input_seq_length'])
train_input_fn = build_lstm_input_fn(x, y, params, is_training=True)
estimator.train(train_input_fn)
def run_seq():
model_fn = build_seq_model_fn(params)
estimator = tf.estimator.Estimator(model_fn=model_fn,
model_dir=params['model_dir'])
examples = pd.read_csv(params['calendar_data']).T
data = examples.values[1:, :]
features = Create_features()(examples)
train_input_fn = build_seq_input_fn(data,
features,
params,
is_training=True)
estimator.train(train_input_fn)
if __name__ == '__main__':
params.update({'lstm_hidden_size': 100})
run_seq()
def adj_obj_margin_func():
params.update({'obj_margin' : max(75, min(250, adjs['delta_obj_margin'] + params['obj_margin']))})
obj_margin_tag.set_text('Object Margin(px): {}'.format(params['obj_margin']))
obj_margin_tag.__build__()
pygame.time.wait(150)
