我正在使用径向基函数功能进行强化学习,我决定使用sklearn生成这些功能,如下所示:
使用以下工具,我的工作进展顺利:
featurizer = sklearn.pipeline.FeatureUnion([
("rbf1", RBFSampler(gamma=5.0, n_components=50)),
("rbf2", RBFSampler(gamma=2.0, n_components=50)),
("rbf3", RBFSampler(gamma=1.0, n_components=50)),
("rbf4", RBFSampler(gamma=0.5, n_components=50))
])
featurizer.fit(scaler.transform(observation_examples))
def featurise_state(state):
scaled = scaler.transform([state])
featurised = featurizer.transform(scaled)
return featurised[0]
问题是,我希望能够保存模型以供以后分析和使用,但我似乎无法确定如何保存和加载稍后使用的FeatureUnion/管道
有什么建议吗
你可以用泡菜:
使用
dump
将对象featurizer
保存在二进制文件中。 然后您将能够使用load
再次加载它相关问题 更多 >
编程相关推荐