embedding_dim = 256
units = 1024
vocab_inp_size = len(inp_lang.word_index)+1
vocab_tar_size = len(targ_lang.word_index)+1
dataset = tf.data.Dataset.from_tensor_slices((input_tensor_train, target_tensor_train)).shuffle(BUFFER_SIZE)
dataset = dataset.batch(BATCH_SIZE, drop_remainder=True)
example_input_batch, example_target_batch = next(iter(dataset))
example_input_batch.shape, example_target_batch.shape
#==================================DataPreparation=======================================
#==================================Encoder===============================================
from lib.encoder import Encoder
encoder = Encoder(vocab_inp_size, embedding_dim, units, BATCH_SIZE)
# sample input
sample_hidden = encoder.initialize_hidden_state()
sample_output, sample_hidden = encoder(example_input_batch, sample_hidden)
print ('Encoder output shape: (batch size, sequence length, units) {}'.format(sample_output.shape))
print ('Encoder Hidden state shape: (batch size, units) {}'.format(sample_hidden.shape))
#==================================Encoder===============================================
#==================================Attention=============================================
from lib.attention import BahdanauAttention
attention_layer = BahdanauAttention(10)
attention_result, attention_weights = attention_layer(sample_hidden, sample_output)
# == ENCODER ==
# Encoder sensor A
pinEncA = 26
# Encoder sensor B
#pinEncB = 20
# == GYROSCOPE ==
# Gyroscope uses pins 2 and 3
# = SETUP SENSORS =
ultraA = Ultra.Measurement(trigA, echoA)
#ultraB = Ultra.Measurement(trigB, echoB)
#ultraC = Ultra.Measurement(trigC, echoC)
encA = Encoder(pinEncA)
#encB = Encoder(pinEncB)
gyroA = Gyroscope()
# = AUXILIARY FUNCTIONS =
def ultraA_read():
tpp = time()
while (True):
raw = ultraA.raw_distance(sample_size=1, sample_wait=0.1)
metric = ultraA.distance_metric(raw)
#sleep(1)
if (time() - tpp > 1):
tpp = time()
from lib.helper import Helper
from lib.encoder import Encoder
import sys
if __name__ == "__main__":
helper = Helper(sys.argv)
helper.ParseConfig().ShowConfigDescription()
encoder = Encoder(helper)
encoder.append_vars(helper.GetConfig("vars"))
template = helper.LoadFile(helper.GetConfig("template"))
data = template
for evasion in helper.GetConfig("evasion"):
data += helper.LoadFile("templates/evasions/" + evasion + ".vba")
data = encoder.replace_var(data, "offset", encoder.get_encoding_offset())
data = encoder.encode_user_vars(data)
data = encoder.append_def_use_tag(data)
data = encoder.rand_vars(data)
data = encoder.rand_int(data)
data = encoder.rand_smallint(data)
encodedvars = helper.GetConfig("encodedvars")
for var in encodedvars:
data = encoder.replace_var(data, var, encodedvars[var], True)
if "-s" in sys.argv or "--split_strings" in sys.argv:
data = encoder.split_strings(data)
if "-x" in sys.argv or "--strings_to_hex" in sys.argv:
