从同一笔记本内不同单元格中的类创建类对象

我需要在不同的单元格中创建一个类的对象

我在 google colab（jupter notebook）中有一个带有 tensorflow 的分类项目。我有一个主类，它的 tensorflow 模型类的对象位于不同的单元格中。我该怎么做呢？

在单元格 1 中，我有：

class Model:

在单元格 2 中，我有：

import Model  # doesn't work 

class Mainprogram:
.
.
.

    def create_model(self):
        self.model = Model(self._Learning_rate, self._epochs, 
                           np.shape(program.train_x),
                           10, [32, 64, 128], strides, pool_size)

program = Mainprogram()
strides = (1, 1)
pool_size = 2
program.train_x, program.train_y, _ = program.read_data()
program.create_model("./train", 0)

我希望它导入所说的类，但它看起来像错误的方法来做任何帮助？