def create_train_params(complete_episodes, complete_steps, success_count, source, normalize, num_episodes, num_steps, time_bonus, wall_bonus, door_bonus, success_bonus):
    """
    Create training parameters dict based on args
    """
    train_params = {}
    # train_params["date_time"] = date_time
    train_params['complete_episodes'] = complete_episodes
    train_params['complete_steps'] = complete_steps
    train_params['success_count'] = success_count
    train_params['source'] = source
    train_params['normalize'] = normalize
    train_params["num_episodes"] = num_episodes
    train_params["num_steps"] = num_steps
    train_params["time_bonus"] = time_bonus
    train_params["wall_bonus"] = wall_bonus
    train_params["door_bonus"] = door_bonus
    train_params["success_bonus"] = success_bonus

    return train_params