我已经使用sklow中的tensorforest以及以下模型文件构建了一个模型
- 检查点
- 图.pbtxt
- 模型.ckpt-1.meta
- model.ckpt-1-00000-of-00001
- events.out.tfevents.1495890133.COGCONSs-MacBook-Pro.local
谁能告诉我如何编写代码来冻结和倾斜模型?我是否必须使用占位符,就像它在 tensorflow 中所做的那样,还是 skflow 不同?这是代码
y_train = pd.factorize(df_train['data.0.sleeping'])[0]
y_test = pd.factorize(df_test['data.0.sleeping'])[0]
df_test=df_test.iloc[:, :-1]
print(df_test.head(3))
hparams = tf.contrib.tensor_forest.python.tensor_forest.ForestHParams(
num_trees=3, max_nodes=1000, num_classes=2, num_features=4)
clf = tf.contrib.learn.TensorForestEstimator(hparams, model_dir='model/')
clf.fit(x=df_train[features], y=y_train,steps=100)