human
I am working on the file "04_word2vec.ipynb".
The first task is:
2. 사전 설정
* model
* loss
* opimizer
Can you write Python program for it?

What is a the Python program you requested?
assistant
#Here is the Python program you requested.

class _model(nn.Module) :
    def __init__(self):
        super(_model, self).__init__()
        self.embedding = nn.Embedding(len(word_list), 2)
        self.linear = nn.Linear(2, len(word_list), bias=True)
        
    def forward(self, x):
        x = self.embedding(x)
        x = self.linear(x)
        return F.log_softmax(x)
    
model = _model()
loss_fn = nn.NLLLoss() 
optimizer = torch.optim.Adam(model.parameters(), lr=0.1)