def render_state_file_text(self):
status = TrainingStatus.NEW
if self.model_to_load is not None:
status = TrainingStatus.NEW_LOAD_MODEL
with open(resource_path('templates/state.yml.mustache'), 'r') as f:
return (chevron.render(f, {
'status': status,
'iterations_run': 0,
'latest_model_saved': ''
}))
def create_project_from_files(input_files):
for file in input_files:
file_name = os.path.basename(file)
if file_name == ".gitignore":
continue
new_project_name = os.path.splitext(file_name)[0]
new_project_dir = project_dir + "/" + to_run_dir + "/" + new_project_name
pathlib.Path(new_project_dir).mkdir(exist_ok=True)
os.rename(file, new_project_dir + "/" + new_project_name + ".txt")
shutil.copyfile(resource_path("templates/config_defaults.yaml"),
new_project_dir + "/templates.yaml")
def render_config_file_text(self):
with open(resource_path('templates/config.yml.mustache'), 'r') as f:
config = Config(
training_file=self.training_file,
output_file="",
output_dir="",
num_loops=self.number_of_iterations.get(),
priority="0",
temperatures_to_generate=list(map(str.strip, self.temperatures_to_generate.get().split(','))),
project_root_dir="",
job_name="",
input_folder="",
output_folder="",
items_to_generate_each_generation=self.items_to_generate_between_generations.get(),
items_to_generate_at_end=self.items_to_generate_at_end.get(),
generate_every_n_generations=self.generation_frequency.get(),
save_model_every_n_generations=self.save_frequency.get(),
dropout=self.dropout.get(),
training_data_percent=self.training_data_percent.get(),
initial_model_to_load=self.model_to_load
)
return config.render(f)
def render_state_file_text(self) -> str:
with open(resource_path('templates/state.yml.mustache'), 'r') as f:
return self.state.render(f)