mirror of
https://github.com/sbrl/research-rainfallradar
synced 2024-11-22 01:12:59 +00:00
Starbeamrainbowlabs
0761651ccf
it should tie into Tensorflow's logging just fine so long as it's the first callback in the queue.
***** TEST SCRIPT *****
model = tf.keras.Sequential([
tf.keras.layers.Dense(64, activation='relu', input_shape=(10,)),
tf.keras.layers.Dense(64, activation='relu'),
tf.keras.layers.Dense(1)
])
model.compile(optimizer='adam', loss='mse', metrics=['mae'])
X = np.random.random((100, 10))
y = np.random.random((100, 1))
split = 80
X_train, X_val = X[:split], X[split:]
y_train, y_val = y[:split], y[split:]
train_dataset = tf.data.Dataset.from_tensor_slices((X_train, y_train)).batch(10)
val_dataset = tf.data.Dataset.from_tensor_slices((X_val, y_val)).batch(10)
history = model.fit(train_dataset,
epochs=10,
validation_data=val_dataset,
callbacks=[
CallbackExtraValidation({
"test": val_dataset
}, verbose=0),
tf.keras.callbacks.CSVLogger("/dev/stdout", separator="\t")
],
verbose=0
)
print(f"DEBUG history {history}")
|
||
|---|---|---|
| .. | ||
| src | ||
| pyproject.toml | ||
| requirements.txt | ||
| slurm-encoderonly-rainfall.job | ||
| slurm-pretrain-plot.job | ||
| slurm-pretrain-predict.job | ||
| slurm-pretrain.job | ||
| slurm-TEST-deeplabv3p-rainfall.job | ||
| slurm-TEST-deeplabv3p.job | ||
| slurm-train-mono.job | ||
| slurm-train.job | ||
| uv.lock | ||