main.py

# Author: Gaël LINKEU
# Purpose: Master Thesis
# Year: 2023


# Iteratively execution of the approach
# 1- Initialize the machine and get the datasets
# 2- Train the RNN
# 3- Construct and merge the prefix tree

import os
import argparse
import random


def parse_args():
    parser = argparse.ArgumentParser()
    parser.add_argument("--n_machines", type=int, default=1)
    parser.add_argument("--times", type=int, default=1)
    return parser.parse_args()

# Study the quality of the finally machine per similarity threshold
def threshold_study():
    args = parse_args()

    sim_thresholds = [0.2, 0.4, 0.6, 0.8, 0.9, 0.95]

    n_machines = args.n_machines
    for j in range(args.times):
        print(f'\n\n The times: {j+1}\n\n')
        for i in range(n_machines):
            id = i
            times = j
            os.system(f'python fsm_initialization.py --id={id} --n_states={id+2}')
            os.system(f'python train_rnn.py --id={id}')
            for sim_threshold in sim_thresholds:
                os.system(f'python extract_mealy.py --id={id} --sim_threshold={sim_threshold}')

if __name__ == "__main__":
    args = parse_args()

    os.system(f'python fsm_initialization.py --id={0}')
    os.system(f'python train_rnn.py --id={0}')
    os.system(f'python extract_mealy.py --id={0}')

    # Experiment multiples times on the same dataset

    # n_machines = args.n_machines
    # for j in range(1):
    #     for i in range(n_machines):
    #         os.system(f'python fsm_initialization.py --id={i} --n_states={i+2}')
    #         os.system(f'python train_rnn.py --id={i} --times={j+1}')
    #         os.system(f'python extract_mealy.py --id={i} --times={j+1}')


    # Experiment one time

    # for i in range(10):
    #     os.system(f'python fsm_initialization.py --id={i}')
    #     os.system(f'python train_rnn.py --id={i}')
    #     os.system(f'python extract_mealy.py --id={i}')


    #Experiment multiple times on different datasets

    # for i in range(10):
    #     os.system(f'python fsm_initialization.py --id={i} --static=0')
    #     os.system(f'python train_rnn.py --id={i}')
    #     os.system(f'python extract_mealy.py --id={i}')