In [1]:
import numpy as np
import torch
import pandas as pd
import matplotlib.pyplot as plt
import seaborn as sns
import pickle
import os
from tqdm.notebook import tqdm
In [2]:
from environment_setup import ScheduleEnv, df, NUM_ACTIVITIES, NUM_TIME_SLOTS
--------------------------------------------------------------------------- ImportError Traceback (most recent call last) Cell In[2], line 1 ----> 1 from environment_setup import ScheduleEnv, df, NUM_ACTIVITIES, NUM_TIME_SLOTS ImportError: cannot import name 'df' from 'environment_setup' (c:\Users\Abdel\Desktop\Assistant-Personnel-pour-la-Gestion-du-Temps\notebooks\environment_setup.py)
In [ ]:
# 2. Définir à nouveau la classe DQNetwork pour pouvoir charger le modèle
class DQNetwork(torch.nn.Module):
def __init__(self, input_dim, output_dim):
super(DQNetwork, self).__init__()
self.fc1 = torch.nn.Linear(input_dim, 128)
self.relu = torch.nn.ReLU()
self.fc2 = torch.nn.Linear(128, 64)
self.fc3 = torch.nn.Linear(64, output_dim)
def forward(self, state):
x = self.relu(self.fc1(state))
x = self.relu(self.fc2(x))
return self.fc3(x)
In [ ]:
# 3. Charger le modèle entraîné et ses paramètres
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
print(f"Utilisation de l'appareil: {device}")
Utilisation de l'appareil: cpu
In [ ]:
# Charger les paramètres du modèle
models_dir = "../models"
with open(os.path.join(models_dir, "model_params.pkl"), "rb") as f:
params = pickle.load(f)
--------------------------------------------------------------------------- FileNotFoundError Traceback (most recent call last) Cell In[16], line 3 1 # Charger les paramètres du modèle 2 models_dir = "../models" ----> 3 with open(os.path.join(models_dir, "model_params.pkl"), "rb") as f: 4 params = pickle.load(f) File c:\Users\Abdel\anaconda3\Lib\site-packages\IPython\core\interactiveshell.py:326, in _modified_open(file, *args, **kwargs) 319 if file in {0, 1, 2}: 320 raise ValueError( 321 f"IPython won't let you open fd={file} by default " 322 "as it is likely to crash IPython. If you know what you are doing, " 323 "you can use builtins' open." 324 ) --> 326 return io_open(file, *args, **kwargs) FileNotFoundError: [Errno 2] No such file or directory: '../models\\model_params.pkl'
In [ ]: